Daniel Gil-U Fuhge commited on
Commit
3ffbd28
1 Parent(s): d9c6096

adjust model path

Browse files
Files changed (1) hide show
  1. 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("data/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,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)