rairo commited on
Commit
c968cc6
·
verified ·
1 Parent(s): eb50f34

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -50,7 +50,7 @@ def bot():
50
  #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
51
  #df['Profit'] = df['Profit'].apply(lambda x: "R{:.1f}".format((x)))
52
  #df['Revenue'] = df['Revenue'].apply(lambda x: "R{:.1f}".format((x)))
53
- llm = ChatGoogleGenerativeAI(model='gemini-1.5-pro', temperature=0.1)
54
  #llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
55
 
56
  agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
 
50
  #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
51
  #df['Profit'] = df['Profit'].apply(lambda x: "R{:.1f}".format((x)))
52
  #df['Revenue'] = df['Revenue'].apply(lambda x: "R{:.1f}".format((x)))
53
+ llm = ChatGoogleGenerativeAI(model='gemini-1.5-pro', temperature=0.1, google_api_key=gemini_api_key)
54
  #llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
55
 
56
  agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)