Spaces:
Runtime error
Runtime error
Daniel Gil-U Fuhge
commited on
Commit
·
3ffbd28
1
Parent(s):
d9c6096
adjust model path
Browse files- animationPipeline.py +3 -3
animationPipeline.py
CHANGED
|
@@ -27,7 +27,7 @@ def animateLogo(path : str):
|
|
| 27 |
use_positional_encoder=True
|
| 28 |
).to(device)
|
| 29 |
|
| 30 |
-
model.load_state_dict(torch.load("
|
| 31 |
|
| 32 |
df = compute_embedding(path, "src/preprocessing/deepsvg/deepsvg_models/deepSVG_hierarchical_ordered.pth.tar")
|
| 33 |
df = df.drop("animation_id", axis=1)
|
|
@@ -46,5 +46,5 @@ def animateLogo(path : str):
|
|
| 46 |
print(result, path)
|
| 47 |
animate_logo(result, path)
|
| 48 |
|
| 49 |
-
logo = "data/examples/logo_181.svg"
|
| 50 |
-
animateLogo(logo)
|
|
|
|
| 27 |
use_positional_encoder=True
|
| 28 |
).to(device)
|
| 29 |
|
| 30 |
+
model.load_state_dict(torch.load("models/animation_transformer.pth"))
|
| 31 |
|
| 32 |
df = compute_embedding(path, "src/preprocessing/deepsvg/deepsvg_models/deepSVG_hierarchical_ordered.pth.tar")
|
| 33 |
df = df.drop("animation_id", axis=1)
|
|
|
|
| 46 |
print(result, path)
|
| 47 |
animate_logo(result, path)
|
| 48 |
|
| 49 |
+
#logo = "data/examples/logo_181.svg"
|
| 50 |
+
#animateLogo(logo)
|