sztanki commited on
Commit
57ab410
·
1 Parent(s): 5584f54

Change zombie model repo

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ transform = transforms.Compose(
89
 
90
 
91
  #ZOMBIE
92
- modelzombie = hf_hub_download(repo_id="akhaliq/JoJoGAN-jojo", filename="jojo_preserve_color.pt")
93
  ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
94
  generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
95
 
 
89
 
90
 
91
  #ZOMBIE
92
+ modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="jojo_preserve_color.pt")
93
  ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
94
  generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
95