sztanki commited on
Commit
27a57d1
1 Parent(s): 271f7dc

Rename model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ transform = transforms.Compose(
80
 
81
 
82
  #ZOMBIE
83
- modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="jojo_zombie_preserve_color.pt")
84
  ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
85
  generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
86
 
 
80
 
81
 
82
  #ZOMBIE
83
+ modelzombie = hf_hub_download(repo_id="Awesimo/jojogan-zombie", filename="zombie.pt")
84
  ckptzombie = torch.load(modelzombie, map_location=lambda storage, loc: storage)
85
  generatorzombie.load_state_dict(ckptzombie["g"], strict=False)
86