Sw1ft0 commited on
Commit
23901ad
·
1 Parent(s): 414da23

Fix Gemini model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ tools = [
60
  )
61
  ]
62
 
63
- llm = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0) # Requires GOOGLE_API_KEY in environment
64
  agent = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)
65
 
66
  def chat_agent(message, history):
 
60
  )
61
  ]
62
 
63
+ llm = ChatGoogleGenerativeAI(model="gemini-2.5-pro", temperature=0) # Requires GOOGLE_API_KEY in environment
64
  agent = initialize_agent(tools, llm, agent="zero-shot-react-description", verbose=True)
65
 
66
  def chat_agent(message, history):