Fix filename
Browse files
app.py
CHANGED
|
@@ -80,7 +80,7 @@ transform = transforms.Compose(
|
|
| 80 |
)
|
| 81 |
|
| 82 |
#HULK
|
| 83 |
-
modelhulk = hf_hub_download(repo_id="Awesimo/jojogan-hulk", filename="
|
| 84 |
ckpthulk = torch.load(modelhulk, map_location=lambda storage, loc: storage)
|
| 85 |
generatorhulk.load_state_dict(ckpthulk["g"], strict=False)
|
| 86 |
|
|
|
|
| 80 |
)
|
| 81 |
|
| 82 |
#HULK
|
| 83 |
+
modelhulk = hf_hub_download(repo_id="Awesimo/jojogan-hulk", filename="hulk.pt")
|
| 84 |
ckpthulk = torch.load(modelhulk, map_location=lambda storage, loc: storage)
|
| 85 |
generatorhulk.load_state_dict(ckpthulk["g"], strict=False)
|
| 86 |
|