Spaces:

dejinlee
/

art

Sleeping

dejinlee commited on Nov 8, 2022

Commit

ea4ade0

1 Parent(s): ef22a91

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 def draw(sets):
-    pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1").to("cuda")
     image = pipe(prompt, guidance_scale=10).images[0]
     return image

 from diffusers import StableDiffusionPipeline
 def draw(sets):
+    pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-EN-v0.1")
     image = pipe(prompt, guidance_scale=10).images[0]
     return image