Update app.py

app.py

@@ -7,14 +7,17 @@ os.system("pip install opencv-python")
 
 if not os.path.exists("data"):
     os.mkdir("data")
-if not os.path.exists("dataout"):
-    os.mkdir("dataout")
+if not os.path.exists("results"):
+    os.mkdir("results")
 
 
 # os.rename("003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth", "experiments/pretrained_models/003_realSR_BSRGAN_DFOWMFC_s64w8_SwinIR-L_x4_GAN.pth")
 def infer(img):
     #img = ImageOps.contain(img, (700, 700))
     width, height = img.size
+    res=np.ones_like(imag.shape)
+    print(img.shape)
+    print(width)
     img.save("./data/data.png")
     os.system('python main_test_swinir.py')
     res=Image.open("./results/data.png")