npc0 commited on
Commit
2a8225f
1 Parent(s): 484c597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -2,7 +2,6 @@ import os
2
  import streamlit as st
3
  import pandas as pd
4
  from pandasai import Agent
5
- from pandasai.responses.streamlit_response import StreamlitResponse
6
  from langchain_groq import ChatGroq
7
 
8
  llm = ChatGroq(
@@ -46,10 +45,7 @@ url = (
46
  if st.session_state['pre_url'] != url:
47
  agent = Agent(
48
  pd.read_csv(url, index_col=0),
49
- config={
50
- "llm": llm,
51
- "response_parser": StreamlitResponse
52
- }
53
  )
54
  st.session_state['agent'] = agent
55
  st.session_state['pre_url'] = url
 
2
  import streamlit as st
3
  import pandas as pd
4
  from pandasai import Agent
 
5
  from langchain_groq import ChatGroq
6
 
7
  llm = ChatGroq(
 
45
  if st.session_state['pre_url'] != url:
46
  agent = Agent(
47
  pd.read_csv(url, index_col=0),
48
+ config={"llm": llm}
 
 
 
49
  )
50
  st.session_state['agent'] = agent
51
  st.session_state['pre_url'] = url