rairo commited on
Commit
2e17513
·
verified ·
1 Parent(s): a36fb46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import streamlit as st
2
  import pandas as pd
3
- from pandasai import SmartDataLake
4
  from pandasai.responses.response_parser import ResponseParser
5
  from pandasai.llm import GoogleGemini
6
  import plotly.express as px
@@ -181,7 +181,7 @@ class StreamLitResponse(ResponseParser):
181
  def generateResponse(prompt, data):
182
  """Generate response using PandasAI with SmartDataLake"""
183
  llm = GoogleGemini(api_key=gemini_api_key)
184
- pandas_agent = SmartDataLake(
185
  data.values(),
186
  config={
187
  "llm": llm,
 
1
  import streamlit as st
2
  import pandas as pd
3
+ from pandasai import SmartDatalake
4
  from pandasai.responses.response_parser import ResponseParser
5
  from pandasai.llm import GoogleGemini
6
  import plotly.express as px
 
181
  def generateResponse(prompt, data):
182
  """Generate response using PandasAI with SmartDataLake"""
183
  llm = GoogleGemini(api_key=gemini_api_key)
184
+ pandas_agent = SmartDatalake(
185
  data.values(),
186
  config={
187
  "llm": llm,