algomuffin commited on
Commit
e758fa3
1 Parent(s): 5e4f4cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -165,8 +165,8 @@ generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
165
 
166
  modeldisneynoc = hf_hub_download(repo_id="algomuffin/disney", filename="disney.pt")
167
 
168
- #ckptMy = torch.load(modelMy, map_location=lambda storage, loc: storage)
169
- #generatorMy.load_state_dict(ckptMy["g"], strict=False)
170
 
171
 
172
 
 
165
 
166
  modeldisneynoc = hf_hub_download(repo_id="algomuffin/disney", filename="disney.pt")
167
 
168
+ ckptMy = torch.load(modeldisneynoc, map_location=lambda storage, loc: storage)
169
+ generatordisneynoc.load_state_dict(ckptMy["g"], strict=False)
170
 
171
 
172