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 |
|