Update app.py

app.py

@@ -78,7 +78,7 @@ def warpImage(im, vx, vy, cast_uint8=True):
     xx, yy = np.meshgrid(x, y)
     XX, YY = np.meshgrid(X, Y)
     #XX = XX + vx
-    XX = np.concatenate([XX, vx], axis = 1)
+    XX = np.concatenate([XX, vx], axis = 0)
     #YY = YY + vy
     YY = np.concatenate([YY, vy], axis = 1)
     mask = (XX < 1) | (XX > width2) | (YY < 1) | (YY > height2)
@@ -207,12 +207,12 @@ def infer():
     transform = T.ToPILImage()
     
     # convert the tensor to PIL image using above transform
-    img = transform(img1_batch[0])
+    img = transform(frames[100])
     
     # display the PIL image
     #img.show()
     img.save('frame_input.jpg')
-    #res = get_warp_res("frame_input.jpg", predicted_flow, 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'
 ####################################