rairo commited on
Commit
7509df8
1 Parent(s): ca5ac0d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -93,7 +93,7 @@ llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=g_api_key, tempe
93
 
94
  def generateResponse(prompt):
95
  llm = GoogleGemini(api_key=g_api_key)
96
- pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts": True,"response_parser":FlaskResponse, "enable_cache": False, "conversational":True})
97
  answer = pandas_agent.chat(prompt)
98
  return answer
99
 
 
93
 
94
  def generateResponse(prompt):
95
  llm = GoogleGemini(api_key=g_api_key)
96
+ pandas_agent = Agent(df,config={"llm":llm, "save_charts_path": user_defined_path,"save_charts": False,"response_parser":FlaskResponse, "enable_cache": False, "conversational":True})
97
  answer = pandas_agent.chat(prompt)
98
  return answer
99