Spaces:
Runtime error
Runtime error
Fixed dict access in app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ models_and_paths = {"akhaliq/JoJoGAN_e4e_ffhq_encode": "e4e_ffhq_encode.pt",
|
|
35 |
"akhaliq/jojogan-stylegan2-ffhq-config-f": f"{model_dir}/base.pt"}
|
36 |
|
37 |
def get_models():
|
38 |
-
for repo_id, file_path in models_and_paths:
|
39 |
hf_hub_download(repo_id=repo_id, filename=file_path)
|
40 |
|
41 |
model_list = ['base'] + [Path(model_ckpt).stem for model_ckpt in os.listdir(model_dir) if not 'base' in model_ckpt]
|
|
|
35 |
"akhaliq/jojogan-stylegan2-ffhq-config-f": f"{model_dir}/base.pt"}
|
36 |
|
37 |
def get_models():
|
38 |
+
for repo_id, file_path in models_and_paths.items():
|
39 |
hf_hub_download(repo_id=repo_id, filename=file_path)
|
40 |
|
41 |
model_list = ['base'] + [Path(model_ckpt).stem for model_ckpt in os.listdir(model_dir) if not 'base' in model_ckpt]
|