Perth0603 commited on
Commit
9df8bf6
·
verified ·
1 Parent(s): c4d989f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", "/mnt/data")
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"