cwadayi commited on
Commit
4c3e3fa
·
verified ·
1 Parent(s): 536891c

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +45 -42
config.py CHANGED
@@ -1,43 +1,46 @@
1
- # config.py
2
- import os
3
- import tempfile
4
- from datetime import datetime
5
-
6
- # --- 環境設定 ---
7
- # 設定 Matplotlib 與 Hugging Face 模型的快取目錄
8
- os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
9
- os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/huggingface")
10
- os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
11
-
12
- # --- LINE Bot 憑證 ---
13
- CHANNEL_ACCESS_TOKEN = os.getenv("CHANNEL_ACCESS_TOKEN")
14
- CHANNEL_SECRET = os.getenv("CHANNEL_SECRET")
15
-
16
- # --- Hugging Face Space URL ---
17
- HF_SPACE_URL = os.getenv("SPACEURL")
18
- if not HF_SPACE_URL:
19
- sid = os.getenv("SPACE_ID")
20
- if sid and "/" in sid:
21
- author, name = sid.split("/", 1)
22
- HF_SPACE_URL = f"https://{author.replace('_', '-')}-{name.replace('_', '-')}.hf.space"
23
- else:
24
- HF_SPACE_URL = ""
25
-
26
- # --- 靜態檔案目錄 ---
27
- STATIC_DIR = os.getenv("STATIC_DIR", os.path.join(tempfile.gettempdir(), "static"))
28
- os.makedirs(STATIC_DIR, exist_ok=True)
29
-
30
- # --- API 端點 ---
31
- CWA_ALARM_API = "https://app-2.cwa.gov.tw/api/v1/earthquake/alarm/list"
32
- USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
33
-
34
- # --- AI 模型設定 ---
35
- LLM_DEVICE = os.getenv("LLM_DEVICE")
36
- LLM_THREADS = os.getenv("LLM_THREADS", "1")
37
- LLM_MODEL = os.getenv("LLM_MODEL", "ckiplab/gpt2-base-chinese")
38
- LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "120"))
39
- LLM_TOP_K = int(os.getenv("LLM_TOP_K", "50"))
40
- LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.7"))
41
-
42
- # --- 顯示用當年年份 ---
 
 
 
43
  CURRENT_YEAR = datetime.now().year
 
1
+ # config.py
2
+ import os
3
+ import tempfile
4
+ from datetime import datetime
5
+
6
+ # --- 環境設定 ---
7
+ # 設定 Matplotlib 與 Hugging Face 模型的快取目錄
8
+ os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
9
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/huggingface")
10
+
11
+ # [修正] 將環境變數的值指派給一個 Python 變數,使其可以被 import
12
+ TRANSFORMERS_CACHE = os.environ["TRANSFORMERS_CACHE"]
13
+ os.makedirs(TRANSFORMERS_CACHE, exist_ok=True)
14
+
15
+ # --- LINE Bot 憑證 ---
16
+ CHANNEL_ACCESS_TOKEN = os.getenv("CHANNEL_ACCESS_TOKEN")
17
+ CHANNEL_SECRET = os.getenv("CHANNEL_SECRET")
18
+
19
+ # --- Hugging Face Space URL ---
20
+ HF_SPACE_URL = os.getenv("SPACEURL")
21
+ if not HF_SPACE_URL:
22
+ sid = os.getenv("SPACE_ID")
23
+ if sid and "/" in sid:
24
+ author, name = sid.split("/", 1)
25
+ HF_SPACE_URL = f"https://{author.replace('_', '-')}-{name.replace('_', '-')}.hf.space"
26
+ else:
27
+ HF_SPACE_URL = ""
28
+
29
+ # --- 靜態檔案目錄 ---
30
+ STATIC_DIR = os.getenv("STATIC_DIR", os.path.join(tempfile.gettempdir(), "static"))
31
+ os.makedirs(STATIC_DIR, exist_ok=True)
32
+
33
+ # --- API 端點 ---
34
+ CWA_ALARM_API = "https://app-2.cwa.gov.tw/api/v1/earthquake/alarm/list"
35
+ USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
36
+
37
+ # --- AI 模型設定 ---
38
+ LLM_DEVICE = os.getenv("LLM_DEVICE")
39
+ LLM_THREADS = os.getenv("LLM_THREADS", "1")
40
+ LLM_MODEL = os.getenv("LLM_MODEL", "ckiplab/gpt2-base-chinese")
41
+ LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "120"))
42
+ LLM_TOP_K = int(os.getenv("LLM_TOP_K", "50"))
43
+ LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.7"))
44
+
45
+ # --- 顯示用當年年份 ---
46
  CURRENT_YEAR = datetime.now().year