Update app.py

app.py

@@ -18,6 +18,11 @@ from transformers import AutoModel, AutoTokenizer
 # 导入预训练模型和分词器
 model_name = "microsoft/phi-2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+# 设置填充令牌，如果分词器没有默认的填充令牌
+if tokenizer.pad_token is None:
+    tokenizer.pad_token = tokenizer.eos_token
+
 model = AutoModel.from_pretrained(model_name)
 
 # Import data