Hellowish commited on
Commit
e113804
·
verified ·
1 Parent(s): 0d587cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -3,8 +3,15 @@ import joblib
3
  import numpy as np
4
  import re
5
  import nltk
 
6
 
7
- nltk.download('punkt', quiet=True)
 
 
 
 
 
 
8
 
9
  # 載入你訓練好的 .pkl 模型
10
  model = joblib.load("ai_detector_model.pkl") # 請確認檔名正確(在 /app 資料夾裡)
 
3
  import numpy as np
4
  import re
5
  import nltk
6
+ import os
7
 
8
+ # 指定 NLTK 資料下載路徑,避免找不到 punkt_tab
9
+ nltk_data_dir = "/usr/local/nltk_data"
10
+ os.makedirs(nltk_data_dir, exist_ok=True)
11
+ nltk.data.path.append(nltk_data_dir)
12
+
13
+ # 下載 punkt tokenizer
14
+ nltk.download('punkt', download_dir=nltk_data_dir, quiet=True)
15
 
16
  # 載入你訓練好的 .pkl 模型
17
  model = joblib.load("ai_detector_model.pkl") # 請確認檔名正確(在 /app 資料夾裡)