Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 from PIL import Image, ImageOps
 import numpy as np
 os.system("pip install opencv-python")
-import cv2
+
 if not os.path.exists("data"):
     os.mkdir("data")
 if not os.path.exists("dataout"):
@@ -17,7 +17,7 @@ def infer(img):
     width, height = img.size
     img.save("./data/data.png")
     os.system('python main_test_swinir.py')
-    res=cv2.imread("./results/data.png")
+    res=Image.open("./results/data.png")
     return res
     #return "butterfly.png"