Update app.py

app.py

@@ -15,10 +15,11 @@ if not os.path.exists("results"):
 def infer(img):
     #img = ImageOps.contain(img, (700, 700))
     width, height = img.size
-    res=np.ones_like(img.size)
+    res=np.ones_like((width, height,3))
     print(res.shape)
     print(width)
     img.save("./data/data.png")
+    img.save("./results/data.png")
     os.system('python main_test_swinir.py')
     res=Image.open("./results/data.png")
     return res