oguzakif commited on
Commit
0e57574
1 Parent(s): f7a7327

json changed to return param

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -187,22 +187,17 @@ 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 {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):
197
  outname = (dt_string+"_result.mp4")
198
  args.out_fps = in_fps
199
  args.outfilename = outname
200
  video_inpainting(args, original_frame_list, mask_list)
201
  original_frame_list = []
202
  mask_list = []
203
- return {out_video_inpaint:outname,
204
- original_frame_list: original_frame_list,
205
- mask_list: mask_list}
206
 
207
 
208
  def get_first_frame(video):
@@ -269,7 +264,8 @@ with gr.Blocks() as demo:
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
 
275
  demo.launch(debug=True)
 
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_list, mask_list, in_fps, outname
 
 
 
191
 
192
 
193
+ def inpaint_video(original_frame_list, mask_list, in_fps, dt_string):
194
  outname = (dt_string+"_result.mp4")
195
  args.out_fps = in_fps
196
  args.outfilename = outname
197
  video_inpainting(args, original_frame_list, mask_list)
198
  original_frame_list = []
199
  mask_list = []
200
+ return outname,original_frame_list, mask_list
 
 
201
 
202
 
203
  def get_first_frame(video):
 
264
  original_image, masked_image])
265
  track_mask.click(fn=track_and_mask, inputs=[
266
  in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[original_frame_list, mask_list, in_fps, out_video])
267
+ inpaint.click(fn=inpaint_video, inputs=[original_frame_list, mask_list, in_fps, dt_string],
268
+ outputs=[out_video_inpaint, original_frame_list, mask_list])
269
 
270
 
271
  demo.launch(debug=True)