DrishtiSharma commited on
Commit
f74dcce
·
verified ·
1 Parent(s): 6cff255

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -109,7 +109,7 @@ if "df" in st.session_state:
109
  df.to_csv("patent_data.csv", index=False)
110
 
111
  csv_agent = create_csv_agent(
112
- ChatOpenAI(temperature=0, model="gpt-4", api_key=st.secrets["api_key"]),
113
  path=["patent_data.csv"],
114
  verbose=True,
115
  agent_type=AgentType.OPENAI_FUNCTIONS,
@@ -131,7 +131,7 @@ if "df" in st.session_state:
131
  chunk = df.iloc[start:start + max_rows]
132
  chunk.to_csv("chunk_data.csv", index=False)
133
  partial_agent = create_csv_agent(
134
- ChatOpenAI(temperature=0, model="gpt-4", api_key=st.secrets["api_key"]),
135
  path=["chunk_data.csv"],
136
  verbose=True,
137
  agent_type=AgentType.OPENAI_FUNCTIONS,
 
109
  df.to_csv("patent_data.csv", index=False)
110
 
111
  csv_agent = create_csv_agent(
112
+ ChatOpenAI(temperature=0, model="gpt-4", api_key=os.getenv("OPENAI_API_KEY")),
113
  path=["patent_data.csv"],
114
  verbose=True,
115
  agent_type=AgentType.OPENAI_FUNCTIONS,
 
131
  chunk = df.iloc[start:start + max_rows]
132
  chunk.to_csv("chunk_data.csv", index=False)
133
  partial_agent = create_csv_agent(
134
+ ChatOpenAI(temperature=0, model="gpt-4", api_key=os.getenv("OPENAI_API_KEY")),
135
  path=["chunk_data.csv"],
136
  verbose=True,
137
  agent_type=AgentType.OPENAI_FUNCTIONS,