Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
| 8 |
|
| 9 |
# ====== 模型来源 ======
|
| 10 |
# 默认从本地目录加载(你上传的文件在 /mnt/data)
|
| 11 |
-
MODEL_DIR = os.environ.get("MODEL_DIR", "/
|
| 12 |
|
| 13 |
# 可选:当模型没写清标签且为二分类时,强制指定顺序(这里通常不需要)
|
| 14 |
# 例:FORCE_BINARY_MAPPING="LEGIT,PHISH" 或 "PHISH,LEGIT"
|
|
|
|
| 8 |
|
| 9 |
# ====== 模型来源 ======
|
| 10 |
# 默认从本地目录加载(你上传的文件在 /mnt/data)
|
| 11 |
+
MODEL_DIR = os.environ.get("MODEL_DIR", "Perth0603/phishing-email-mobilebert")
|
| 12 |
|
| 13 |
# 可选:当模型没写清标签且为二分类时,强制指定顺序(这里通常不需要)
|
| 14 |
# 例:FORCE_BINARY_MAPPING="LEGIT,PHISH" 或 "PHISH,LEGIT"
|