Update app.py

app.py

@@ -1,18 +1,22 @@
-import torch
+import os
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
 import gradio as gr
 
-# 模型 ID - 你在 Hugging Face 上的模型路径
-repo_id = "huohuo828/bishe"  # 你的模型ID
+# 从环境变量中获取 Token
+token = os.getenv("HF_TOKEN")  # 获取在 Hugging Face Secrets 中设置的 "HF_TOKEN"
 
-# 加载模型和tokenizer
-tokenizer = AutoTokenizer.from_pretrained(repo_id)
-model = AutoModelForSequenceClassification.from_pretrained(repo_id)
+# 模型 ID
+repo_id = "huohuo828/bishe"
+
+# 加载模型和 tokenizer
+tokenizer = AutoTokenizer.from_pretrained(repo_id, use_auth_token=token)
+model = AutoModelForSequenceClassification.from_pretrained(repo_id, use_auth_token=token)
 
 # 处理输入的函数
 def predict(symptoms):
     try:
-        # 对输入的症状进行tokenize
+        # 对输入的症状进行 tokenization
         inputs = tokenizer(symptoms, return_tensors="pt", padding=True, truncation=True, max_length=512)
         with torch.no_grad():
             # 模型推理
@@ -39,3 +43,4 @@ gr.Interface(
     description="通过输入症状，预测可能的疾病和推荐科室。",
 ).launch(debug=True)
 
+