Fix model loading with weights_only=False

app.py

@@ -1,20 +1,20 @@
 """
-Hugging Face Spaces 入口文件
+Hugging Face Spaces 入口
 """
 import sys
 
-# 补丁 1: audioop (Python 3.13)
+# 补丁: audioop (Python 3.13)
 try:
     import audioop_lts as audioop
     sys.modules["pyaudioop"] = audioop
 except ImportError:
     pass
 
-# 补丁 2: HfFolder (huggingface-hub>=0.25)
+# 补丁: HfFolder
 try:
     import huggingface_hub
     if not hasattr(huggingface_hub, "HfFolder"):
-        from huggingface_hub import HfApi, get_token
+        from huggingface_hub import get_token
         class _HfFolder:
             @staticmethod
             def get_token(): return get_token()
@@ -23,4 +23,6 @@ try:
 except Exception:
     pass
 
+# 启动
 from webui import demo
+demo.queue().launch(server_name="0.0.0.0", server_port=7860)

predict.py

@@ -44,7 +44,7 @@ class GarbageClassifier:
         if not path.exists():
             raise FileNotFoundError(f"模型文件不存在: {model_path}\n请先运行 train.py 训练模型")
 
-        checkpoint = torch.load(model_path, map_location=self.device, weights_only=True)
+        checkpoint = torch.load(model_path, map_location=self.device, weights_only=False)
         model.load_state_dict(checkpoint["model_state_dict"])
         model = model.to(self.device)
         print(f"✓ 模型加载成功 ({model_path})")