Hellowish commited on
Commit
2b36831
·
verified ·
1 Parent(s): e113804

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -5,13 +5,18 @@ 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 資料夾裡)
 
5
  import nltk
6
  import os
7
 
8
+ # 建議用使用者目錄確保可寫入權限
9
+ nltk_data_dir = os.path.join(os.path.expanduser("~"), "nltk_data")
10
  os.makedirs(nltk_data_dir, exist_ok=True)
11
+
12
+ # 將資料夾加入 nltk 搜尋路徑
13
  nltk.data.path.append(nltk_data_dir)
14
 
15
+ # 下載 punkt tokenizer,如果已存在則不重複下載
16
+ try:
17
+ nltk.data.find("tokenizers/punkt")
18
+ except LookupError:
19
+ nltk.download("punkt", download_dir=nltk_data_dir, quiet=True)
20
 
21
  # 載入你訓練好的 .pkl 模型
22
  model = joblib.load("ai_detector_model.pkl") # 請確認檔名正確(在 /app 資料夾裡)