RageshAntony commited on
Commit
792ef4e
·
verified ·
1 Parent(s): a6fe1b1
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -243,10 +243,12 @@ def process_video_frames(
243
  )
244
  print_image_info(result_image)
245
  save_generated_image(result_image,index)
246
- yield result_image,result_image
 
247
  processed_frames.append(result_image)
248
-
249
- yield processed_frames
 
250
 
251
 
252
  @spaces.GPU(duration=120)
@@ -370,7 +372,8 @@ def submit_function_flux(
370
  new_result_image = Image.new("RGB", (width + condition_width + 5, height))
371
  new_result_image.paste(conditions, (0, 0))
372
  new_result_image.paste(result_image, (condition_width + 5, 0))
373
- return new_result_image, result_image
 
374
 
375
 
376
  def person_example_fn(image_path):
 
243
  )
244
  print_image_info(result_image)
245
  save_generated_image(result_image,index)
246
+ gallery_images = update_gallery()
247
+ yield result_image,gallery_images
248
  processed_frames.append(result_image)
249
+ gallery_images = update_gallery()
250
+ gallery_images = update_gallery()
251
+ yield processed_frames, gallery_images
252
 
253
 
254
  @spaces.GPU(duration=120)
 
372
  new_result_image = Image.new("RGB", (width + condition_width + 5, height))
373
  new_result_image.paste(conditions, (0, 0))
374
  new_result_image.paste(result_image, (condition_width + 5, 0))
375
+ gallery_images = update_gallery()
376
+ return new_result_image, gallery_images
377
 
378
 
379
  def person_example_fn(image_path):