Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
4fc4e60
1
Parent(s):
8f2e8f4
Update inference_realesrgan.py
Browse files- inference_realesrgan.py +2 -2
inference_realesrgan.py
CHANGED
|
@@ -64,9 +64,9 @@ def main():
|
|
| 64 |
netscale = 4
|
| 65 |
|
| 66 |
# determine model paths
|
| 67 |
-
model_path = os.path.join('
|
| 68 |
if not os.path.isfile(model_path):
|
| 69 |
-
model_path = os.path.join('
|
| 70 |
if not os.path.isfile(model_path):
|
| 71 |
raise ValueError(f'Model {args.model_name} does not exist.')
|
| 72 |
|
|
|
|
| 64 |
netscale = 4
|
| 65 |
|
| 66 |
# determine model paths
|
| 67 |
+
model_path = os.path.join('.', args.model_name + '.pth')
|
| 68 |
if not os.path.isfile(model_path):
|
| 69 |
+
model_path = os.path.join('.', args.model_name + '.pth')
|
| 70 |
if not os.path.isfile(model_path):
|
| 71 |
raise ValueError(f'Model {args.model_name} does not exist.')
|
| 72 |
|