test gradio

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
+from diffusers import StableDiffusion3ControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 from huggingface_hub import login
 import os
 import spaces
@@ -14,7 +14,7 @@ controlnet_id = "lllyasviel/control_v11p_sd15_inpaint"
 
 # Load the ControlNet model and Stable Diffusion pipeline
 controlnet = ControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float16)
-pipe = StableDiffusionControlNetPipeline.from_pretrained(
+pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
     model_id, controlnet=controlnet, torch_dtype=torch.float16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)