Update app.py
Browse files
app.py
CHANGED
|
@@ -5,13 +5,18 @@ import re
|
|
| 5 |
import nltk
|
| 6 |
import os
|
| 7 |
|
| 8 |
-
#
|
| 9 |
-
nltk_data_dir = "
|
| 10 |
os.makedirs(nltk_data_dir, exist_ok=True)
|
|
|
|
|
|
|
| 11 |
nltk.data.path.append(nltk_data_dir)
|
| 12 |
|
| 13 |
-
# 下載 punkt tokenizer
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
| 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 資料夾裡)
|