Spaces:
Runtime error
Runtime error
algomuffin
commited on
Commit
•
498bf42
1
Parent(s):
2fe33c2
Update app.py
Browse files
app.py
CHANGED
@@ -163,7 +163,7 @@ ckptsketch = torch.load(modelSketch, map_location=lambda storage, loc: storage)
|
|
163 |
generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
|
164 |
|
165 |
|
166 |
-
|
167 |
|
168 |
#ckptMy = torch.load(modelMy, map_location=lambda storage, loc: storage)
|
169 |
#generatorMy.load_state_dict(ckptMy["g"], strict=False)
|
|
|
163 |
generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
|
164 |
|
165 |
|
166 |
+
modelMy = hf_hub_download(repo_id="algomuffin/my_model", filename="my.pt")
|
167 |
|
168 |
#ckptMy = torch.load(modelMy, map_location=lambda storage, loc: storage)
|
169 |
#generatorMy.load_state_dict(ckptMy["g"], strict=False)
|