Update app.py

app.py

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