Rename model
Browse files
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="
|
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 |
|