Update app.py

app.py

@@ -55,7 +55,11 @@ def load_models():
 
         #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"
+        # 使用 huggingface_hub 下载模型文件
+        sketch_weight_path = hf_hub_download(
+            repo_id="OmniSVG/OmniSVG-3B",
+            filename="pytorch_model.bin",
+            cache_dir="./models")
         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()