Shuang59 commited on
Commit
204dc42
β€’
1 Parent(s): 65e6336

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -23,6 +23,7 @@ from diffusers import StableDiffusionPipeline
23
 
24
  has_cuda = th.cuda.is_available()
25
  device = th.device('cpu' if not th.cuda.is_available() else 'cuda')
 
26
 
27
  # init stable diffusion model
28
  pipe = StableDiffusionPipeline.from_pretrained(
@@ -154,7 +155,7 @@ iface = gr.Interface(compose,
154
  gr.Slider(2, 30),
155
  gr.Slider(10, 200)
156
  ],
157
- outputs='image',
158
  title=title, description=description, examples=examples)
159
 
160
  iface.launch()
 
23
 
24
  has_cuda = th.cuda.is_available()
25
  device = th.device('cpu' if not th.cuda.is_available() else 'cuda')
26
+ print(device)
27
 
28
  # init stable diffusion model
29
  pipe = StableDiffusionPipeline.from_pretrained(
 
155
  gr.Slider(2, 30),
156
  gr.Slider(10, 200)
157
  ],
158
+ outputs='image', cache_examples=False,
159
  title=title, description=description, examples=examples)
160
 
161
  iface.launch()