Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import gradio as gr
 from PIL import Image
 import torch
+import shutil
 
 model_path = 'experiments/pretrained_models/Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.pth'
 
@@ -16,16 +17,15 @@ else:
     print(f'downloading model {model_path}')
     open(model_path, 'wb').write(r.content)
 
-os.makedirs("test", exist_ok=True)
-    
+
+os.makedirs("test", exist_ok=True)    
+
 def inference(img):
-    
-    cv2.imwrite("test/1.png", cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
-    # basewidth = 256
-    # wpercent = (basewidth/float(img.size[0]))
-    # hsize = int((float(img.size[1])*float(wpercent)))
-    # img = img.resize((basewidth,hsize), Image.ANTIALIAS)
-    #img.save("test/1.jpg", "JPEG")
+    basewidth = 256
+    wpercent = (basewidth/float(img.size[0]))
+    hsize = int((float(img.size[1])*float(wpercent)))
+    img = img.resize((basewidth,hsize), Image.ANTIALIAS)
+    img.save("test/1.png", "PNG")
     os.system('python main_test_swin2sr.py --task real_sr --model_path experiments/pretrained_models/Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.pth --folder_lq test --scale 4')
     return 'results/swin2sr_real_sr_x4/1_Swin2SR.png'
         
@@ -36,7 +36,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2209.113
 examples=[['butterflyx4.png']]
 gr.Interface(
     inference, 
-    "image", 
+    gr.inputs.Image(type="pil", label="Input"),
     "image",
     title=title,
     description=description,