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():
|