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()
|