oguzakif commited on
Commit
f7a7327
1 Parent(s): 2a1896f

output order changed

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -187,10 +187,10 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
187
  outname = (dt_string+"_output.avi")
188
  print('Original Frame Count: ',len(original_list))
189
  print('Mask Frame Count: ',len(mask_list))
190
- return {out_video:outname,
191
- original_frame_list: original_list,
192
  mask_list: mask_list,
193
- in_fps: in_fps}
 
194
 
195
 
196
  def inpaint_video(original_frame_list, mask_list):
@@ -268,7 +268,7 @@ with gr.Blocks() as demo:
268
  approve_mask.click(lambda x: [x['image'], x['mask']], first_frame, [
269
  original_image, masked_image])
270
  track_mask.click(fn=track_and_mask, inputs=[
271
- in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[out_video, original_frame_list, mask_list, in_fps])
272
  inpaint.click(fn=inpaint_video, outputs=[out_video_inpaint, original_frame_list, mask_list])
273
 
274
 
 
187
  outname = (dt_string+"_output.avi")
188
  print('Original Frame Count: ',len(original_list))
189
  print('Mask Frame Count: ',len(mask_list))
190
+ return {original_frame_list: original_list,
 
191
  mask_list: mask_list,
192
+ in_fps: in_fps,
193
+ out_video:outname}
194
 
195
 
196
  def inpaint_video(original_frame_list, mask_list):
 
268
  approve_mask.click(lambda x: [x['image'], x['mask']], first_frame, [
269
  original_image, masked_image])
270
  track_mask.click(fn=track_and_mask, inputs=[
271
+ in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[original_frame_list, mask_list, in_fps, out_video])
272
  inpaint.click(fn=inpaint_video, outputs=[out_video_inpaint, original_frame_list, mask_list])
273
 
274