update deepsvg model

animationPipeline.py

@@ -29,7 +29,7 @@ def animateLogo(path : str):
 
     model.load_state_dict(torch.load("models/animation_transformer.pth", map_location=torch.device('cpu')), strict=False)
 
-    df = compute_embedding(path, "src/preprocessing/deepsvg/deepsvg_models/deepSVG_hierarchical_ordered.pth.tar")
+    df = compute_embedding(path, "models/deepSVG_hierarchical_ordered.pth.tar")
     df = df.drop("animation_id", axis=1)
 
     df = pd.concat([df, pd.DataFrame(0, index=df.index, columns=range(df.shape[1], df.shape[1] + 26))], axis=1, ignore_index=True).astype(float)

models/deepSVG_hierarchical_ordered.pth.tar

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b44e93095426b18ef2dfaf5ebbced8b7d29dcd6ce79325681d976acb1366a22
+size 41280531

src/preprocessing/preprocessing.py

@@ -4,7 +4,6 @@ import copy
 import torch
 import glob
 import pandas as pd
-import streamlit as st
 import pickle
 from xml.dom import minidom
 from svgpathtools import svg2paths2
@@ -61,7 +60,6 @@ def compute_embedding(path: str, model_path: str, save: str = None) -> pd.DataFr
     decomposed_svgs = {}
     
     for i in range(len(paths)):
-        st.write(document)
         doc_temp = copy.deepcopy(document)
         paths_temp = doc_temp.getElementsByTagName('path')
         current_path = paths_temp[i]