Update app.py

app.py

@@ -77,8 +77,10 @@ def warpImage(im, vx, vy, cast_uint8=True):
     Y = np.linspace(1, height1, height1)
     xx, yy = np.meshgrid(x, y)
     XX, YY = np.meshgrid(X, Y)
-    XX = XX + vx
-    YY = YY + vy
+    #XX = XX + vx
+    XX = np.concatenate([XX, vx])
+    #YY = YY + vy
+    YY = np.concatenate([YY, vy])
     mask = (XX < 1) | (XX > width2) | (YY < 1) | (YY > height2)
     XX = np.clip(XX, 1, width2)
     YY = np.clip(XX, 1, height2)