Spaces:
Running
Running
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)
|