end
Browse files
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 = "
|
| 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
|