Update app.py

app.py

@@ -87,10 +87,17 @@ def get_frames(video_in, step, name, resize_w):
 
 def create_video(frames, fps, type):
     print("building video result")
+    #https://zulko.github.io/moviepy/getting_started/videoclips.html#mask-clips
+    
     clip = ImageSequenceClip(frames, fps=fps)
-    clip.write_gif(type + "_result.gif", fps=fps)
     
-    return type + "_result.gif"
+    maskclip = ImageSequenceClip(frames, fps=fps) #, ismask=True)
+    maskclip.to_mask()
+    clip.set_mask(maskclip)
+    
+    clip.write_videofile(type + "_result.mp4", fps=fps)
+    
+    return type + "_result.mp4"
 
     
 def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
@@ -241,9 +248,9 @@ with gr.Blocks() as demo:
                 submit_btn = gr.Button("Submit")
             
             with gr.Column():
-                video_output = gr.Image()
+                video_output = gr.Video()
                 file_output = gr.File()
-                depth_output = gr.Image()
+                depth_output = gr.Video()
     
     gr.Examples(
         examples=[["./examples/streetview.mp4", 1, 0, 256, 15, True]],