Spaces:

OmniSVG
/

OmniSVG-3B

Paused

OmniSVG commited on 26 days ago

Commit

1245532

verified ·

1 Parent(s): c3d8327

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,7 +57,7 @@ def load_models():
         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()
-        svg_tokenizer = SVGTokenizer('https://huggingface.co/OmniSVG/OmniSVG/resolve/main/config.yaml')
 def process_and_resize_image(image_input, target_size=(200, 200)):

         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()
+        svg_tokenizer = SVGTokenizer('config.yaml')
 def process_and_resize_image(image_input, target_size=(200, 200)):