AlanOC commited on
Commit
fe81190
·
verified ·
1 Parent(s): 5887012

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-20240229':
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-20240229'
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
  ]