Update app.py

app.py

@@ -203,12 +203,13 @@ def infer():
     flo_file = write_flo(predicted_flow, "flofile.flo")
     #write_jpeg(frames[100], f"input_image.jpg")
     #res = warp_image(img1_batch, predicted_flow)
+    my_frame = transforms(F.resize(frames[100], size=[520, 960]))
     
     # define a transform to convert a tensor to PIL image
     transform = T.ToPILImage()
     
     # convert the tensor to PIL image using above transform
-    img = transform(frames[100])
+    img = transform(my_frame)
     
     # display the PIL image
     #img.show()