Update app.py

app.py

@@ -149,6 +149,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
           alpha = 1.0/len(result_frames)
           beta = (1.0 - alpha)
           res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
+          depth_frames.append(rgb)
         break
       nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
       if o_flow == True:
@@ -161,14 +162,14 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
       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)
+      depth_frames.append(rgb)
       prvs = nxt
 
     print("averaged frames: " + str(len(result_frames)))
-    depth_vid = create_video(depth_frames, fps_output, "depth map")
+    depth_vid = create_video(depth_frames, fps_output, "depth_map")
     cv2.imwrite('opticalfb.png', res)
     depth_map.release()
     cv2.destroyAllWindows()