Update app.py

app.py

@@ -12,7 +12,7 @@ import glob
 
 
 from decoder import SketchDecoder
-from transformers import AutoTokenizer, AutoProcessor
+from transformers import AutoTokenizer, AutoProcessor, Qwen2_5_VLForConditionalGeneration
 from qwen_vl_utils import process_vision_info
 from tokenizer import SVGTokenizer
 
@@ -52,8 +52,9 @@ def load_models():
         processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-VL-3B-Instruct", padding_side="left")
 
         sketch_decoder = SketchDecoder()
-        
-        sketch_weight_path = "https://huggingface.co/OmniSVG/OmniSVG/resolve/main/OmniSVG-3B.bin"
+
+        sketch_weight_path = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG")
+        #sketch_weight_path = "https://huggingface.co/OmniSVG/OmniSVG/resolve/main/OmniSVG-3B.bin"
         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()