Spaces:

IDEA-CCNL
/

Taiyi-Stable-Diffusion-Chinese

Runtime error

weifeng-chen commited on Nov 18, 2022

Commit

2278ab0

•

1 Parent(s): 8a3cc0d

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,8 +16,8 @@ def resize(w_val,l_val,img):
 def infer(prompt, guide, steps, width, height):
-    images = pipe_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
-    image = images[0]
     return image
 gr.Interface(fn=infer, inputs=

 def infer(prompt, guide, steps, width, height):
+    output = pipe_text2img(prompt, guidance_scale=guide, num_inference_steps=steps, width=width, height=height)
+    image = images.images[0]
     return image
 gr.Interface(fn=infer, inputs=