prithivMLmods commited on
Commit
5187002
·
verified ·
1 Parent(s): ec32fc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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