Update app.py

app.py

@@ -102,7 +102,7 @@ def get_warp_res(fname_image, fname_flow, fname_output='warped.png'):
     print(f"FNAME IMAGE: {fname_image}")
     im2 = imread(fname_image)
     print(f"FNAME IMAGE READED: {im2}")
-    flow = readFlowFile(fname_flow)
+    flow = fname_flow
     im_warped, _ = warpImage(im2, flow[:, :, 0], flow[:, :, 1])
     imwrite(fname_output, im_warped)
     
@@ -206,7 +206,7 @@ def infer():
     # display the PIL image
     #img.show()
     img.save('frame_input.jpg')
-    res = get_warp_res("frame_input.jpg", "flofile.flo", fname_output='warped.png')
+    res = get_warp_res("frame_input.jpg", predicted_flow, fname_output='warped.png')
     print(res)
     return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
 ####################################