Update app.py

app.py

@@ -6,7 +6,7 @@ def load_model():
     # 在这里加载 Hugging Face 模型
     # 比如使用 transformers 库来加载模型
     from transformers import AutoModelForSequenceClassification, AutoTokenizer
-    model_name = "zeroMN/zeroMN_SHMT"
+    model_name = "https://huggingface.co/zeroMN/SHMT"
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     return model, tokenizer
@@ -14,7 +14,7 @@ def load_model():
 def infer(text):
     model, tokenizer = load_model()
     inputs = tokenizer(text, return_tensors="pt")
-    outputs = model("zeroMN/zeroMN_SHMT")
+    outputs = model("https://huggingface.co/zeroMN/SHMT")
     # 根据模型输出生成响应
     return f"Predicted response for: {text}"