use interface

app.py

@@ -4,9 +4,9 @@ from diffusers import StableDiffusionPipeline
 
 
 device="cuda"
-model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1"
 
-pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id).to(device)
+# pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id).to(device)
+io1_text2img = gr.Interface.load("models/IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1")
 
 def resize(w_val,l_val,img):
   img = Image.open(img)
@@ -15,7 +15,8 @@ def resize(w_val,l_val,img):
 
 
 def infer(prompt, guide, steps, width, height): 
-    output = pipe_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
+    
+    output = io1_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
     image = output.images[0]
     return image