Update app.py

app.py

@@ -53,7 +53,7 @@ def load_models():
 
         sketch_decoder = SketchDecoder()
 
-        sketch_weight_path = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG")
+        sketch_weight_path = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG-3B")
         #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()