Hu commited on
Commit
d701b84
·
1 Parent(s): 15d0c07

change loading image

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -125,9 +125,9 @@ model.eval()
125
  # return grid
126
 
127
 
128
- def super_reso(image):
129
  # gradio open image as np array
130
- # image = Image.fromarray(image, mode="RGB") # not needed, gradio already processes it
131
 
132
  # prediction
133
  with torch.no_grad():
 
125
  # return grid
126
 
127
 
128
+ def super_reso(image_path):
129
  # gradio open image as np array
130
+ image = Image.fromarray(image_path, mode="RGB") # not needed, gradio already processes it
131
 
132
  # prediction
133
  with torch.no_grad():