Spaces:
Running
Running
Hu
commited on
Commit
·
d701b84
1
Parent(s):
15d0c07
change loading image
Browse files
app.py
CHANGED
@@ -125,9 +125,9 @@ model.eval()
|
|
125 |
# return grid
|
126 |
|
127 |
|
128 |
-
def super_reso(
|
129 |
# gradio open image as np array
|
130 |
-
|
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():
|