Spaces:

dejinlee
/

art

Sleeping

dejinlee commited on Nov 9, 2022

Commit

81d446b

1 Parent(s): b4c4699

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline
 def draw(sets):
     pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1")
-    images[] = pipe(sets, guidance_scale=10).images[]
     return images[0],images[1],images[2],images[3],images[4]
 iface = gr.Interface(fn=draw, inputs="text", outputs=["image","image","image","image","image"],)

 def draw(sets):
     pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1")
+    images = pipe(sets, guidance_scale=10).images[]
     return images[0],images[1],images[2],images[3],images[4]
 iface = gr.Interface(fn=draw, inputs="text", outputs=["image","image","image","image","image"],)