Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -234,8 +234,8 @@ def generate(
|
|
| 234 |
torch.cuda.empty_cache()
|
| 235 |
|
| 236 |
selected_pipe = models.get(model_choice, pipe)
|
| 237 |
-
images = selected_pipe(**options).images
|
| 238 |
yield "⚡ Generating image ███████▒▒▒ 69%"
|
|
|
|
| 239 |
image_path = save_image(images[0])
|
| 240 |
yield gr.Image(image_path)
|
| 241 |
return
|
|
|
|
| 234 |
torch.cuda.empty_cache()
|
| 235 |
|
| 236 |
selected_pipe = models.get(model_choice, pipe)
|
|
|
|
| 237 |
yield "⚡ Generating image ███████▒▒▒ 69%"
|
| 238 |
+
images = selected_pipe(**options).images
|
| 239 |
image_path = save_image(images[0])
|
| 240 |
yield gr.Image(image_path)
|
| 241 |
return
|