Ahsen Khaliq commited on
Commit
4fc4e60
1 Parent(s): 8f2e8f4

Update inference_realesrgan.py

Browse files
Files changed (1) hide show
  1. 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('experiments/pretrained_models', args.model_name + '.pth')
68
  if not os.path.isfile(model_path):
69
- model_path = os.path.join('realesrgan/weights', args.model_name + '.pth')
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