Update README.md

README.md

@@ -1,3 +1,56 @@
 ---
 license: apache-2.0
+datasets:
+- detection-datasets/coco
+language:
+- en
+library_name: diffusers
+tags:
+- pytorch
+- controlnet
+- image-colorization
+- image-to-image
+pipeline_tag: image-to-image
 ---
+
+# Model Card for ColorizeNet
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+This model is a ControlNet training to perform image colorization from black and white images.
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+ColorizeNet is an image colorization model based on ControlNet, trained using the pre-trained Stable Diffusion model version 2.1 proposed by Stability AI.
+
+- **Finetuned from model :** [https://huggingface.co/stabilityai/stable-diffusion-2-1]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [https://github.com/rensortino/ColorizeNet]
+- 
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+The model has been trained on COCO, using all the images in the dataset and converting them to grayscale to use them to condition the ControlNet
+
+[https://huggingface.co/datasets/detection-datasets/coco]
+
+### Results
+
+[More Information Needed]