Update app.py

app.py

@@ -8,10 +8,10 @@ import gradio as gr
 # 取得 Hugging Face Token，用於 API 調用推論服務
 from huggingface_hub import InferenceClient
 
-HF_TOKEN = os.environ.get("GROQ_API_KEY")
+API_KEY = os.environ.get("GROQ_API_KEY")
 API_BASE_URL = os.environ.get("GROQ_API_BASE_URL")
 client = InferenceClient(
-    api_key=HF_TOKEN,
+    api_key=API_KEY,
     base_url=API_BASE_URL
 )
 
@@ -130,7 +130,7 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
     # 使用 InferenceClient 呼叫 API 模型產生新題目
     completion = client.chat.completions.create(
         # model="mistralai/Mistral-7B-Instruct-v0.3",
-        model="deepseek-r1-distill-qwen-32b",
+        model="gemma2-9b-it",
         # model="mistralai/Mistral-Nemo-Instruct-2407", 
         messages=messages,
         max_tokens=1024