Spaces:
Runtime error
Runtime error
algomuffin
commited on
Commit
·
5e4f4cb
1
Parent(s):
498bf42
Update app.py
Browse files
app.py
CHANGED
@@ -96,7 +96,7 @@ generatorspider = deepcopy(original_generator)
|
|
96 |
|
97 |
generatorsketch = deepcopy(original_generator)
|
98 |
|
99 |
-
|
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 |
-
|
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)
|