denaneek commited on
Commit
2310272
·
1 Parent(s): 13376d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -1,14 +1,9 @@
1
  from langchain.embeddings.openai import OpenAIEmbeddings
2
- from langchain.vectorstores import Chroma
3
- from langchain.text_splitter import CharacterTextSplitter
4
- from langchain.chains.question_answering import load_qa_chain
5
  from langchain.agents import create_csv_agent
6
  from langchain.llms import OpenAI
7
  import os
8
 
9
- import pandas as pd
10
  file_name = 'gpt-generated-food-data.csv'
11
- df = pd.read_csv(file_name)
12
 
13
  agent = create_csv_agent(OpenAI(temperature=0),
14
  file_name,
@@ -17,8 +12,7 @@ agent = create_csv_agent(OpenAI(temperature=0),
17
  agent.agent.llm_chain.prompt.template
18
 
19
  def make_inference(query):
20
- docs = docsearch.get_relevant_documents(query)
21
- return(agent.run(input_documents=docs, question=query))
22
 
23
  if __name__ == "__main__":
24
  # make a gradio interface
 
1
  from langchain.embeddings.openai import OpenAIEmbeddings
 
 
 
2
  from langchain.agents import create_csv_agent
3
  from langchain.llms import OpenAI
4
  import os
5
 
 
6
  file_name = 'gpt-generated-food-data.csv'
 
7
 
8
  agent = create_csv_agent(OpenAI(temperature=0),
9
  file_name,
 
12
  agent.agent.llm_chain.prompt.template
13
 
14
  def make_inference(query):
15
+ return(agent.run(query))
 
16
 
17
  if __name__ == "__main__":
18
  # make a gradio interface