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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -96,7 +96,7 @@ generatorspider = deepcopy(original_generator)
96
 
97
  generatorsketch = deepcopy(original_generator)
98
 
99
- generatorMy = deepcopy(original_generator)
100
 
101
  transform = transforms.Compose(
102
  [
@@ -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
- 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)
 
96
 
97
  generatorsketch = deepcopy(original_generator)
98
 
99
+ generatordisneynoc = deepcopy(original_generator)
100
 
101
  transform = transforms.Compose(
102
  [
 
163
  generatorsketch.load_state_dict(ckptsketch["g"], strict=False)
164
 
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)