xd11yggy commited on
Commit
80e5262
Β·
verified Β·
1 Parent(s): 83590ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -119,6 +119,7 @@ def respond(
119
  message,
120
  history: list[tuple[str, str]],
121
  system_message,
 
122
  max_tokens,
123
  temperature,
124
  top_p,
@@ -148,7 +149,7 @@ def respond(
148
 
149
  try:
150
  completion = client.chat.completions.create(
151
- model="qwen/qwq-32b:free",
152
  messages=messages,
153
  max_tokens=max_tokens,
154
  temperature=temperature,
@@ -202,6 +203,12 @@ demo = gr.ChatInterface(
202
  gr.Slider(minimum=1000, maximum=15000, value=6000, step=500, label="Max Tokens"),
203
  gr.Slider(minimum=0.1, maximum=1.0, value=0.5, step=0.1, label="Temperature"),
204
  gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="Top-p"),
 
 
 
 
 
 
205
  gr.Textbox(label="OpenRouter API Key", type="password")
206
  ],
207
  title="Web Research Agent πŸ€–",
 
119
  message,
120
  history: list[tuple[str, str]],
121
  system_message,
122
+ model_name,
123
  max_tokens,
124
  temperature,
125
  top_p,
 
149
 
150
  try:
151
  completion = client.chat.completions.create(
152
+ model=model_name,
153
  messages=messages,
154
  max_tokens=max_tokens,
155
  temperature=temperature,
 
203
  gr.Slider(minimum=1000, maximum=15000, value=6000, step=500, label="Max Tokens"),
204
  gr.Slider(minimum=0.1, maximum=1.0, value=0.5, step=0.1, label="Temperature"),
205
  gr.Slider(minimum=0.1, maximum=1.0, value=0.85, step=0.05, label="Top-p"),
206
+ gr.Textbox(
207
+ value="qwen/qwq-32b:free", # Default model
208
+ label="Model",
209
+ placeholder="deepseek/deepseek-r1-zero:free, google/gemini-2.0-pro-exp-02-05:free...",
210
+ info="OpenRouter model ID"
211
+ ),
212
  gr.Textbox(label="OpenRouter API Key", type="password")
213
  ],
214
  title="Web Research Agent πŸ€–",