update rootdir
This commit is contained in:
@@ -90,7 +90,7 @@ def main():
|
|||||||
else:
|
else:
|
||||||
model_path = os.path.join('realesrgan/weights', args.model_name + '.pth')
|
model_path = os.path.join('realesrgan/weights', args.model_name + '.pth')
|
||||||
if not os.path.isfile(model_path):
|
if not os.path.isfile(model_path):
|
||||||
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||||
for url in file_url:
|
for url in file_url:
|
||||||
# model_path will be updated
|
# model_path will be updated
|
||||||
model_path = load_file_from_url(
|
model_path = load_file_from_url(
|
||||||
|
|||||||
Reference in New Issue
Block a user