test gradio

app.py

@@ -3,16 +3,15 @@ import torch
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
 
 # Load Stable Diffusion model and ControlNet reference-only model
-model_id = "runwayml/stable-diffusion-v1-5"
+model_id = "stabilityai/stable-diffusion-3.5-large-turbo"
 controlnet_id = "lllyasviel/control_v11p_sd15_inpaint"  # Use an appropriate ControlNet variant
 
-# Initialize the pipeline on CPU
 controlnet = ControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float32)
 pipeline = StableDiffusionControlNetPipeline.from_pretrained(
     model_id,
     controlnet=controlnet,
     torch_dtype=torch.float32
-).to("cpu")
+)
 
 # Define the Gradio interface function
 def generate_image(prompt, reference_image):