Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|