Spaces:
Runtime error
Runtime error
output order changed
Browse files
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 {
|
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=[
|
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 |
|