rinong commited on
Commit
f3abb0d
·
1 Parent(s): 67ae780

Fixed dict access in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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]