BtB-ExpC commited on
Commit
2899fe9
·
1 Parent(s): a587854

small changes

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. config/llm_config.py +7 -5
app.py CHANGED
@@ -104,7 +104,7 @@ with gr.Blocks() as demo:
104
  with gr.Row():
105
  model_choice = gr.Dropdown(
106
  choices=list(llms.keys()),
107
- value="gpt-4o",
108
  label="Select LLM",
109
  interactive=True,
110
  )
 
104
  with gr.Row():
105
  model_choice = gr.Dropdown(
106
  choices=list(llms.keys()),
107
+ value="GPT-4o",
108
  label="Select LLM",
109
  interactive=True,
110
  )
config/llm_config.py CHANGED
@@ -29,11 +29,13 @@ def create_deepseek_llm(model_name: str, temperature: float):
29
  return ChatAnthropic(api_key=ANTHROPIC_API_KEY, model_name=model_name, temperature=temperature)
30
 
31
  llms = {
32
- "gpt-4o": create_openai_llm("gpt-4o", LOW),
33
- "gpt-4o-mini": create_openai_llm("gpt-4o-mini", LOW),
34
- "gpt-4o_high_temp": create_openai_llm("gpt-4o", HIGH),
35
- "gpt-4o-mini_high_temp": create_openai_llm("gpt-4o-mini", HIGH),
36
- "o1": create_openai_reasoning_llm("o1"),
 
 
37
  "Claude 3.5": create_anthropic_llm("claude-3-5-sonnet-latest", LOW),
38
  "Deepseek R1🚧": create_anthropic_llm("deepseek-reasoner", LOW),
39
  }
 
29
  return ChatAnthropic(api_key=ANTHROPIC_API_KEY, model_name=model_name, temperature=temperature)
30
 
31
  llms = {
32
+ "GPT-4o": create_openai_llm("gpt-4o", LOW),
33
+ "GPT-4o-mini": create_openai_llm("gpt-4o-mini", LOW),
34
+ "GPT-4o_high_temp": create_openai_llm("gpt-4o", HIGH),
35
+ "GPT-4o-mini_high_temp": create_openai_llm("gpt-4o-mini", HIGH),
36
+ "GPT-4 Turbo": create_openai_llm("gpt-4-turbo-2024-04-09", HIGH),
37
+ "o1": create_openai_reasoning_llm("o1-2024-12-17"),
38
+ "o3-mini": create_openai_reasoning_llm("o3-mini-2025-01-31"),
39
  "Claude 3.5": create_anthropic_llm("claude-3-5-sonnet-latest", LOW),
40
  "Deepseek R1🚧": create_anthropic_llm("deepseek-reasoner", LOW),
41
  }