Update app.py

app.py

@@ -53,9 +53,10 @@ def load_models():
 
         sketch_decoder = SketchDecoder()
 
-        sketch_weight_path = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG-3B")
+        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))
+        #sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()
         
         svg_tokenizer = SVGTokenizer('config.yaml')