README.md

---
license: mit
datasets:
- phiyodr/coco2017
---
An implementation of the [Pix2Pix](https://arxiv.org/abs/1611.07004) paper for image colorization using PyTorch.

Try a trained on Google Colab:  

<a href="https://colab.research.google.com/github/DavidRamosArchilla/Colorizing-images/blob/main/image_colorization_demo.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>

---
To download it:

```python
from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="davidramos/image-colorization", 
    filename="model.pth" 
)
```

and to load the weights into a model, first clone this repo (or grab the needed files):
```bash
git clone https://github.com/yourusername/colorizing_images.git
cd colorizing_images
```

and create a model intance:
```python
from models import ColorizingNet
model = ColorizingNet(device)
model.load_state_dict(torch.load(model_path))
```