Update app.py

app.py

@@ -9,8 +9,10 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
+API_BASE_URL = os.environ.get("API_BASE_URL")
 client = InferenceClient(
     api_key=HF_TOKEN
+    base_url=API_BASE_URL
 )
 
 # 載入問答資料庫路徑
@@ -128,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="perplexity-ai/r1-1776",
+        model="nvjob/DeepSeek-R1-32B-Cline:latest",
         # model="mistralai/Mistral-Nemo-Instruct-2407", 
         messages=messages,
         max_tokens=2048