Update app.py

app.py

@@ -9,6 +9,7 @@ import argparse
 import gc
 import yaml
 import glob
+from huggingface_hub import hf_hub_download
 
 
 from decoder import SketchDecoder
@@ -58,8 +59,7 @@ def load_models():
         # 使用 huggingface_hub 下载模型文件
         sketch_weight_path = hf_hub_download(
             repo_id="OmniSVG/OmniSVG-3B",
-            filename="pytorch_model.bin",
-            cache_dir="./models")
+            filename="pytorch_model.bin",)
         sketch_decoder.load_state_dict(torch.load(sketch_weight_path))
         sketch_decoder = sketch_decoder.to(device).eval()