Change zombie model repo
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ transform = transforms.Compose(
|
|
| 89 |
|
| 90 |
|
| 91 |
#ZOMBIE
|
| 92 |
-
modelzombie = hf_hub_download(repo_id="
|
| 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 |
|