Update config.py

config.py

@@ -7,8 +7,6 @@ from datetime import datetime
 # 設定 Matplotlib 與 Hugging Face 模型的快取目錄
 os.environ.setdefault("MPLCONFIGDIR", "/tmp/matplotlib")
 os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/huggingface")
-
-# [FIX] Assign the environment variable's value to a Python variable to make it importable.
 TRANSFORMERS_CACHE = os.environ["TRANSFORMERS_CACHE"]
 os.makedirs(TRANSFORMERS_CACHE, exist_ok=True)
 
@@ -34,13 +32,12 @@ os.makedirs(STATIC_DIR, exist_ok=True)
 CWA_ALARM_API = "https://app-2.cwa.gov.tw/api/v1/earthquake/alarm/list"
 USGS_API_BASE_URL = "https://earthquake.usgs.gov/fdsnws/event/1/query"
 
-# --- AI 模型設定 ---
-LLM_DEVICE = os.getenv("LLM_DEVICE")
-LLM_THREADS = os.getenv("LLM_THREADS", "1")
-LLM_MODEL = os.getenv("LLM_MODEL", "ckiplab/gpt2-base-chinese")
-LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "120"))
+# --- AI 模型設定 (已更新為 Gemma) ---
+HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")  # [新增] 用於讀取 HF Token
+LLM_MODEL = os.getenv("LLM_MODEL", "google/gemma-2b-it")  # [修改] 更新預設模型
+LLM_MAX_NEW_TOKENS = int(os.getenv("LLM_MAX_NEW_TOKENS", "256")) # [修改] 增加 token 數量
 LLM_TOP_K = int(os.getenv("LLM_TOP_K", "50"))
 LLM_TEMPERATURE = float(os.getenv("LLM_TEMPERATURE", "0.7"))
 
 # --- 顯示用當年年份 ---
-CURRENT_YEAR = datetime.now().year
+CURRENT_YEAR = datetime.now().year