Update app.py

app.py

@@ -137,15 +137,17 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
     prvs = cv2.cvtColor(fr1, cv2.COLOR_RGBA2GRAY)
     hsv = np.zeros_like(fr1)
     hsv[..., 1] = 255
-    if o_flow == True:
-      res = np.zeros_like(prvs)
-      flow = res
-    else:
-      res = prvs
+    res = np.zeros_like(prvs)
+    flow = res
         
     while(depth_map.isOpened()):
       ret, fr2 = depth_map.read()
       if ret == False:
+        if o_flow == False:
+          rgb = prvs
+          alpha = 1.0/len(result_frames)
+          beta = (1.0 - alpha)
+          res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
         break
       nxt = cv2.cvtColor(fr2, cv2.COLOR_RGBA2GRAY)
       if o_flow == True:
@@ -156,7 +158,7 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
         rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
         rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
       else:
-        rgb = nxt
+        rgb = prvs
           
       alpha = 1.0/len(result_frames)
       beta = (1.0 - alpha)