Update app.py

app.py

@@ -178,8 +178,8 @@ def infer():
     #print(flow_imgs)
 
     predicted_flow = list_of_flows[-1][0]
-    print(f"predicted flow dtype = {predicted_flows.dtype}")
-    print(f"predicted flow shape = {predicted_flows.shape}")
+    print(f"predicted flow dtype = {predicted_flow.dtype}")
+    print(f"predicted flow shape = {predicted_flow.shape}")
     
     flow_img = flow_to_image(predicted_flow).to("cpu")
 #     output_folder = "/tmp/"  # Update this to the folder of your choice
@@ -206,7 +206,10 @@ def infer():
     print(f"frame1pil shape: {frame1pil.shape}")
     print(f"frame1pil dtype: {frame1pil.dtype}")
     img.save('frame_input.jpg')
-    #res = get_warp_res('frame_input.jpg', "predicted_flow.jpg", 'warped.png')
+    print(f"img1_batch: {img1_batch}")
+    print(f"img1_batch shape: {img1_batch.shape}")
+    print(f"img1_batch dtype: {img1_batch.dtype}")
+    #res = get_warp_res(img1_batch, "predicted_flow.jpg", 'warped.png')
     #print(res)
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
 ####################################