change PIL load image

app.py

@@ -54,8 +54,8 @@ def pred_SRCNN(model, image, device, scale_factor=2):
     model.to(device)
     model.eval()
 
-    # open image
-    # image = Image.open(image_path)
+    # open image, gradio opens image as nparray
+    image = Image.fromarray(image)
     # split channels
     y, cb, cr = image.convert("YCbCr").split()
     # size will be used in image transform
@@ -127,8 +127,8 @@ model.eval()
 
 def super_reso(image_path):
     # gradio open image as np array
-    image_array = np.asarray(image_path)
-    image = Image.fromarray(image_array, mode="RGB") 
+    #image_array = np.asarray(image_path)
+    #image = Image.fromarray(image_array, mode="RGB") 
 
     # prediction
     with torch.no_grad():