Update app.py

app.py

@@ -133,6 +133,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     depth_map = cv2.VideoCapture(final_vid)
     print("interpolated frames: " + str(len(frames_list)) + " -> " + str(len(result_frames)))
 
+    depth_frames = []
     ret, fr1 = depth_map.read()
     prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
     hsv = np.zeros_like(fr1)
@@ -159,18 +160,20 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
         rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
       else:
         rgb = prvs
-          
+
+      depth_frames.append(rgb)
       alpha = 1.0/len(result_frames)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
       prvs = nxt
 
     print("averaged frames: " + str(len(result_frames)))
+    depth_vid = create_video(depth_frames, fps_output, "depth map")
     cv2.imwrite('opticalfb.png', res)
     depth_map.release()
     cv2.destroyAllWindows()
 
-    return final_vid, files, 'opticalfb.png'
+    return final_vid, files, depth_vid
 
 
 def logscale(linear):
@@ -222,7 +225,7 @@ with gr.Blocks() as demo:
             with gr.Column():
                 video_output = gr.Video()
                 file_output = gr.File()
-                depth_output = gr.ImageEditor(image_mode="L", interactive=True, label="Depth map")
+                depth_output = gr.Video()
     
     gr.Examples(
         examples=[["./examples/streetview.mp4", 1, 0, 256, 15, False]],