Update app.py

app.py

@@ -7,8 +7,8 @@ def load_model():
     # 比如使用 transformers 库来加载模型
     from transformers import AutoModelForSequenceClassification, AutoTokenizer
     model_name = "https://huggingface.co/zeroMN/SHMT"
-    model = AutoModelForSequenceClassification.from_pretrained(model_name)
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSequenceClassification.from_pretrained(zeroMN/SHMT)
+    tokenizer = AutoTokenizer.from_pretrained(zeroMN/SHMT)
     return model, tokenizer
 
 def infer(text):