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 |
|