Update app.py

app.py

@@ -159,13 +159,13 @@ def infer(url_in,interpolation,fps_output,resize_n,winsize,o_flow):
         hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)
         rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2RGB)
         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)
+      rgb = cv2.cvtColor(rgb, cv2.COLOR_GRAY2RGB)
       cv2.imwrite(f"opticalfb{i}.jpg", rgb)
       depth_frames.append(f"opticalfb{i}.jpg")
       i+=1