robot4 commited on
Commit
8b90ce7
·
verified ·
1 Parent(s): 9596651
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
  # === 强制指定 BERT 配置 ===
5
  from transformers import BertConfig, BertTokenizer, BertForSequenceClassification
6
  MODEL_REPO = "robot4/emotion"
7
- MODEL_SUBFOLDER = "checkpoints_finally"
8
  print(f"正在加载 BERT 模型: {MODEL_REPO} ...")
9
  try:
10
  # 1. 显式加载 BERT Tokenizer
 
4
  # === 强制指定 BERT 配置 ===
5
  from transformers import BertConfig, BertTokenizer, BertForSequenceClassification
6
  MODEL_REPO = "robot4/emotion"
7
+ MODEL_SUBFOLDER = "checkpoints_finally_end"
8
  print(f"正在加载 BERT 模型: {MODEL_REPO} ...")
9
  try:
10
  # 1. 显式加载 BERT Tokenizer