Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -123,7 +123,7 @@ def get_claude_sonnet():
|
|
123 |
|
124 |
# Function to get the appropriate LLM based on the selected model
|
125 |
def get_llm(model_name, temperature):
|
126 |
-
if model_name == 'claude-3-sonnet-
|
127 |
anthropic_client = get_claude_sonnet()
|
128 |
return AnthropicLLM(client=anthropic_client, model=model_name)
|
129 |
else:
|
@@ -346,7 +346,7 @@ def main():
|
|
346 |
'gpt-3.5-turbo-16k',
|
347 |
'gpt-3.5-turbo-1106',
|
348 |
'gpt-4',
|
349 |
-
'claude-3-sonnet-
|
350 |
|
351 |
# Other custom or fine-tuned models can be added here
|
352 |
]
|
|
|
123 |
|
124 |
# Function to get the appropriate LLM based on the selected model
|
125 |
def get_llm(model_name, temperature):
|
126 |
+
if model_name == 'claude-3-5-sonnet-20240620':
|
127 |
anthropic_client = get_claude_sonnet()
|
128 |
return AnthropicLLM(client=anthropic_client, model=model_name)
|
129 |
else:
|
|
|
346 |
'gpt-3.5-turbo-16k',
|
347 |
'gpt-3.5-turbo-1106',
|
348 |
'gpt-4',
|
349 |
+
'claude-3-5-sonnet-20240620'
|
350 |
|
351 |
# Other custom or fine-tuned models can be added here
|
352 |
]
|