Spaces:

akhaliq
/

JoJoGAN

Runtime error

akhaliq HF staff commited on Feb 17, 2022

Commit

73bc6ed

•

1 Parent(s): 5043e00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,7 +41,7 @@ os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqs
 os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
 device= 'cpu'
-model_path = 'models/e4e_ffhq_encode.pt'
 ckpt = torch.load(model_path, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path

 os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
 device= 'cpu'
+model_path = hf_hub_download(repo_id="akhaliq/JoJoGAN_e4e_ffhq_encode", filename="e4e_ffhq_encode.pt")
 ckpt = torch.load(model_path, map_location='cpu')
 opts = ckpt['opts']
 opts['checkpoint_path'] = model_path