Update README.md

README.md

@@ -27,6 +27,37 @@ Our data can be found in the following link: [RCID Dataset](https://huggingface.
 
 The code is released on [Color Illusion](https://github.com/mao1207/RCID)
 
+## How to Use the Model
+
+To generate a realistic image from a simplified image and a text prompt using the Color Diffusion model, you can use the following code:
+
+```python
+import random
+import torch
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
+from diffusers.utils import load_image
+
+# Set device
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Load the models
+controlnet = ControlNetModel.from_pretrained("controlnet_model_path", torch_dtype=torch.float32).to(device)
+pipe = StableDiffusionControlNetPipeline.from_pretrained("base_model_path", controlnet=controlnet, torch_dtype=torch.float32).to(device)
+
+# Load your simplified image
+simplified_image = load_image("path_to_simplified_image.png")
+
+# Define the text prompt
+prompt = "A photorealistic image of a sunset over the ocean."
+
+# Generate realistic image
+generator = torch.manual_seed(random.randint(0, 100000))
+generated_image = pipe(prompt, num_inference_steps=50, generator=generator, image=simplified_image).images[0]
+
+# Save the generated image
+generated_image.save("generated_image.png")
+‘’‘
+
 ## License
 
 The source code of this repository is released under the Apache License 2.0. The model license and dataset license are listed on their corresponding webpages.