update

app.py

@@ -3,6 +3,7 @@ from transformers import pipeline, AutoTokenizer
 import json
 from onnxruntime import InferenceSession
 import numpy as np
+from huggingface_hub import hf_hub_download
 
 # 设置页面配置
 st.set_page_config(page_title="中文垃圾信息分类器", page_icon="🚫", layout="wide")
@@ -11,11 +12,16 @@ st.set_page_config(page_title="中文垃圾信息分类器", page_icon="🚫", l
 @st.cache_resource
 def load_classifiers():
     hf_classifier = pipeline("text-classification", model="app-x/chinese_spam_classifier")
-    onnx_session = InferenceSession("app-x/chinese_spam_classifier_onnx/model_optimized.onnx")
+    
+    # 从Hugging Face Hub下载ONNX模型
+    onnx_path = hf_hub_download("app-x/chinese_spam_classifier_onnx", "model_optimized.onnx")
+    onnx_session = InferenceSession(onnx_path)
+    
     tokenizer = AutoTokenizer.from_pretrained("app-x/chinese_spam_classifier_onnx")
     
     # 加载配置文件
-    with open("app-x/chinese_spam_classifier_onnx/config.json", "r") as f:
+    config_path = hf_hub_download("app-x/chinese_spam_classifier_onnx", "config.json")
+    with open(config_path, "r") as f:
         config = json.load(f)
     
     id2label = config["id2label"]

requirements.txt

@@ -2,6 +2,7 @@ streamlit
 transformers
 torch
 onnxruntime
+huggingface_hub