Update app.py

app.py

@@ -158,15 +158,16 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
         hsv[..., 0] = ang*180/np.pi/2
         hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
         rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
-        cv2.imwrite(f"opticalfb{i}.jpg", rgb)
-        depth_frames.append(f"opticalfb{i}.jpg")
         rgb = cv2.cvtColor(rgb, cv2.COLOR_RGBA2GRAY)
+        rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGB)
       else:
         rgb = prvs
 
       alpha = 1.0/len(result_frames)
       beta = (1.0 - alpha)
       res = cv2.addWeighted(rgb, alpha, res, beta, 0.0, res)
+      cv2.imwrite(f"opticalfb{i}.jpg", rgb)
+      depth_frames.append(f"opticalfb{i}.jpg")
       i+=1
       prvs = nxt