aicodingfun commited on
Commit
2797f2e
·
verified ·
1 Parent(s): 7df5d92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ import gradio as gr
9
  from huggingface_hub import InferenceClient
10
 
11
  HF_TOKEN = os.environ.get("HF_TOKEN")
12
- API_BASE_URL = os.environ.get("API_BASE_URL")
13
  client = InferenceClient(
14
  api_key=HF_TOKEN,
15
  base_url=API_BASE_URL
@@ -130,7 +130,7 @@ def generate_math_questions(grade, term, qtype="Unspecified", num_questions=10):
130
  # 使用 InferenceClient 呼叫 API 模型產生新題目
131
  completion = client.chat.completions.create(
132
  # model="mistralai/Mistral-7B-Instruct-v0.3",
133
- model="nvjob/DeepSeek-R1-32B-Cline:latest",
134
  # model="mistralai/Mistral-Nemo-Instruct-2407",
135
  messages=messages,
136
  max_tokens=1024
 
9
  from huggingface_hub import InferenceClient
10
 
11
  HF_TOKEN = os.environ.get("HF_TOKEN")
12
+ API_BASE_URL = os.environ.get("GROQ_API_BASE_URL")
13
  client = InferenceClient(
14
  api_key=HF_TOKEN,
15
  base_url=API_BASE_URL
 
130
  # 使用 InferenceClient 呼叫 API 模型產生新題目
131
  completion = client.chat.completions.create(
132
  # model="mistralai/Mistral-7B-Instruct-v0.3",
133
+ model="deepseek-r1-distill-qwen-32b",
134
  # model="mistralai/Mistral-Nemo-Instruct-2407",
135
  messages=messages,
136
  max_tokens=1024