RageshAntony commited on
Commit
1b319b5
·
verified ·
1 Parent(s): 4f5ce17

fix return

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -205,7 +205,7 @@ def extract_frames(video_path):
205
  return frames
206
 
207
  #process_video_frames
208
- @spaces.GPU(duration=120)
209
  def process_video_frames(
210
  video,
211
  cloth_image,
@@ -251,7 +251,7 @@ def process_video_frames(
251
  yield processed_frames, gallery_images
252
 
253
 
254
- @spaces.GPU(duration=120)
255
  def proc_function_vidfl(
256
  person_image,
257
  cloth_image,
@@ -298,7 +298,7 @@ def proc_function_vidfl(
298
  print("YEILEDING proc_function_vidfl")
299
  return result_image
300
 
301
- @spaces.GPU(duration=120)
302
  def submit_function_flux(
303
  person_image,
304
  cloth_image,
@@ -483,7 +483,7 @@ def app_gradio():
483
  submit_function_flux,
484
  [person_image_flux, cloth_image_flux, cloth_type, num_inference_steps_flux, guidance_scale_flux,
485
  seed_flux, show_type],
486
- result_image_flux,gal_output
487
  )
488
 
489
  with gr.Tab("Video Flux"):
@@ -568,7 +568,7 @@ def app_gradio():
568
  process_video_frames,
569
  [person_image_vidflux, cloth_image_vidflux, cloth_type, num_inference_steps_vidflux, guidance_scale_vidflux,
570
  seed_vidflux, show_type],
571
- result_image_vidflux,gal_output
572
  )
573
 
574
 
 
205
  return frames
206
 
207
  #process_video_frames
208
+ @spaces.GPU(duration=175)
209
  def process_video_frames(
210
  video,
211
  cloth_image,
 
251
  yield processed_frames, gallery_images
252
 
253
 
254
+ @spaces.GPU(duration=175)
255
  def proc_function_vidfl(
256
  person_image,
257
  cloth_image,
 
298
  print("YEILEDING proc_function_vidfl")
299
  return result_image
300
 
301
+ @spaces.GPU(duration=175)
302
  def submit_function_flux(
303
  person_image,
304
  cloth_image,
 
483
  submit_function_flux,
484
  [person_image_flux, cloth_image_flux, cloth_type, num_inference_steps_flux, guidance_scale_flux,
485
  seed_flux, show_type],
486
+ [result_image_flux,gal_output]
487
  )
488
 
489
  with gr.Tab("Video Flux"):
 
568
  process_video_frames,
569
  [person_image_vidflux, cloth_image_vidflux, cloth_type, num_inference_steps_vidflux, guidance_scale_vidflux,
570
  seed_vidflux, show_type],
571
+ [result_image_vidflux,gal_output]
572
  )
573
 
574