Spaces:
Running
on
Zero
Running
on
Zero
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
|