Update app.py

app.py

@@ -240,7 +240,7 @@ def infer():
     # convert the tensor to PIL image using above transform
     #img = transform(frames[1])
     img = transform(input_frame_2)
-    img = img.resize((520, 960))
+    img = img.resize((960, 520))
     # display the PIL image
     #img.show()
     frame2pil = np.array(img.convert('RGB'))