Update app.py

app.py

@@ -16,7 +16,7 @@ def infer(img):
     #img = ImageOps.contain(img, (700, 700))
     width, height = img.size
     res=np.ones_like(img.size)
-    print(img.shape)
+    print(res.shape)
     print(width)
     img.save("./data/data.png")
     os.system('python main_test_swinir.py')