robot4 commited on
Commit
debae11
·
verified ·
1 Parent(s): 5442735

还是路径的修改

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -4,7 +4,17 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
5
  # === 关键配置 ===
6
  # 这里直接指向您刚才上传好的模型仓库
7
- MODEL_REPO = "robot4/emotion/checkpoint-6500"
 
 
 
 
 
 
 
 
 
 
8
 
9
  print(f"正在加载模型: {MODEL_REPO} ...")
10
  try:
 
4
 
5
  # === 关键配置 ===
6
  # 这里直接指向您刚才上传好的模型仓库
7
+ # === 改这里 ===
8
+ MODEL_REPO = "robot4/emotion" # 仓库名 (不能带文件夹)
9
+ MODEL_SUBFOLDER = "checkpoint-6500" # 子文件夹名
10
+ print(f"正在加载模型: {MODEL_REPO} (目录: {MODEL_SUBFOLDER})...")
11
+ try:
12
+ # 关键:加上 subfolder 参数!
13
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_REPO, subfolder=MODEL_SUBFOLDER)
14
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_REPO, subfolder=MODEL_SUBFOLDER)
15
+ except Exception as e:
16
+ print(f"Error: {e}")
17
+ raise e
18
 
19
  print(f"正在加载模型: {MODEL_REPO} ...")
20
  try: