Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ if not GOOGLE_API_KEY:
|
|
| 40 |
|
| 41 |
def generateResponse(prompt, dfs):
|
| 42 |
llm = GoogleGemini(api_key=GOOGLE_API_KEY)
|
| 43 |
-
pandas_agent = SmartDataframe(
|
| 44 |
answer = pandas_agent.chat(prompt)
|
| 45 |
return answer
|
| 46 |
|
|
|
|
| 40 |
|
| 41 |
def generateResponse(prompt, dfs):
|
| 42 |
llm = GoogleGemini(api_key=GOOGLE_API_KEY)
|
| 43 |
+
pandas_agent = SmartDataframe(dfs,config={"llm":llm, "response_parser":StreamLitResponse})
|
| 44 |
answer = pandas_agent.chat(prompt)
|
| 45 |
return answer
|
| 46 |
|