Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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)
|