Update app.py

app.py

@@ -54,12 +54,7 @@ def load_models():
 
         sketch_decoder = SketchDecoder()
 
-        #model = Qwen2_5_VLForConditionalGeneration.from_pretrained("OmniSVG/OmniSVG-3B")
-        #print("Loading Successfully!")
-        # 使用 huggingface_hub 下载模型文件
-        sketch_weight_path = hf_hub_download(
-            repo_id="OmniSVG/OmniSVG-3B",
-            filename="pytorch_model.bin",)
+        sketch_weight_path = hf_hub_download(repo_id="OmniSVG/OmniSVG", filename="pytorch_model.bin",)
         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()