Ahsen Khaliq commited on
Commit
3bc45b9
1 Parent(s): 94f8dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ transform = transforms.Compose(
71
  os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
72
  os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
73
 
74
- os.system("gdown https://drive.google.com/uc?id=1ZRwYLRytCEKi__eT2Zxv1IlV6BGVQ_K2")
75
 
76
  ckptjojo = torch.load('jojo_preserve_color.pt', map_location=lambda storage, loc: storage)
77
  generatorjojo.load_state_dict(ckptjojo["g"], strict=False)
 
71
  os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
72
  os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
73
 
74
+ os.system("wget https://huggingface.co/akhaliq/JoJoGAN-jojo/resolve/main/jojo_preserve_color.pt")
75
 
76
  ckptjojo = torch.load('jojo_preserve_color.pt', map_location=lambda storage, loc: storage)
77
  generatorjojo.load_state_dict(ckptjojo["g"], strict=False)