small changes
Browse files- app.py +1 -1
- 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="
|
| 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 |
-
"
|
| 33 |
-
"
|
| 34 |
-
"
|
| 35 |
-
"
|
| 36 |
-
"
|
|
|
|
|
|
|
| 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 |
}
|