Update app.py

app.py

@@ -213,7 +213,7 @@ def infer():
     
     # convert the tensor to PIL image using above transform
     img = transform(frames[100])
-    img = img.resize((520, 960))
+    img = img.resize((960, 520))
     # display the PIL image
     #img.show()
     img.save('frame_input.jpg')