ChrisSacrumCor commited on
Commit
20c6159
·
verified ·
1 Parent(s): d41b8a0

gemini 1.5

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -359,7 +359,7 @@ class FinalAgent:
359
  self.model = LiteLLMModel(
360
  api_key=api_key,
361
  # model_id="gpt-4o",
362
- model_id="gemini/gemini-pro",
363
  # model_id="anthropic/claude-3-5-sonnet-20240620",
364
  temperature=0.5 # Added temperature parameter
365
  )
 
359
  self.model = LiteLLMModel(
360
  api_key=api_key,
361
  # model_id="gpt-4o",
362
+ model_id="gemini/gemini-1.5-flash",
363
  # model_id="anthropic/claude-3-5-sonnet-20240620",
364
  temperature=0.5 # Added temperature parameter
365
  )