sss

app.py

@@ -1,8 +1,19 @@
 import gradio as gr
+import torch
 
 def generate_image(image):
     if image is not None:
-        return image['layers'][0]
+        sketch = image['layers'][0]
+        with torch.no_grad():  # Disable gradient calculation for inference
+            model_output = pipe(prompt, num_inference_steps=20, generator=torch.manual_seed(0), image=sketch)
+            generated_image = model_output.images[0]
+
+    return generated_image
+
+controlnet_model_name_or_path = "controlnet_path"
+pretrained_model_name_or_path = "runwayml/stable-diffusion-v1-5"
+controlnet = ControlNetModel.from_pretrained(controlnet_model_name_or_path, torch_dtype=torch.float16,  conditioning_channels=3)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(pretrained_model_name_or_path, controlnet=controlnet, torch_dtype=torch.float16, safety_checker=None)
 
 iface = gr.Interface(
     fn=generate_image,