Update app.py

app.py

@@ -177,5 +177,28 @@ def infer():
 #
 # ffmpeg -f image2 -framerate 30 -i predicted_flow_%d.jpg -loop -1 flow.gif
 
-
+def warp_flow(img, flow):
+      h, w = flow.shape[:2]
+      flow = flow.copy()
+      flow[:, :, 0] += np.arange(w)
+      flow[:, :, 1] += np.arange(h)[:, np.newaxis]
+      res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
+      return res
+
+  
+
+  def warp(frame1, frame2, flo_path, blend=0.5, weights_path=None):
+    flow21 = np.load(flo_path)
+    frame1pil = np.array(frame1.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
+    frame1_warped21 = warp_flow(frame1pil, flow21)
+    # frame2pil = frame1pil
+    frame2pil = np.array(frame2.convert('RGB').resize((flow21.shape[1],flow21.shape[0])))
+
+    if weights_path: 
+      forward_weights = load_cc(weights_path)
+      blended_w = frame2pil*(1-blend) + blend*(frame1_warped21*forward_weights+frame2pil*(1-forward_weights))
+    else: blended_w = frame2pil*(1-blend) + frame1_warped21*(blend)
+
+    return  PIL.Image.fromarray(blended_w.astype('uint8'))
+      
 gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(), gr.Files()]).launch()