Update app.py

app.py

@@ -53,10 +53,10 @@ def load_models():
 
         sketch_decoder = SketchDecoder()
 
-        model = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG-3B")
-        print("Loading Successfully!")
-        #sketch_weight_path = "https://huggingface.co/OmniSVG/OmniSVG/resolve/main/OmniSVG-3B.bin"
-        #sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
+        #model = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG-3B")
+        #print("Loading Successfully!")
+        sketch_weight_path = "https://huggingface.co/OmniSVG/OmniSVG-3B/resolve/main/pytorch_model.bin"
+        sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()
         
         svg_tokenizer = SVGTokenizer('config.yaml')