shuhulhandoo commited on
Commit
063d1c6
1 Parent(s): 8d40917

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from langchain.embeddings.openai import OpenAIEmbeddings
3
  from langchain.text_splitter import CharacterTextSplitter
4
- from langchain.vectorstores import Chroma
5
 
6
  from langchain.chains import ConversationalRetrievalChain
7
  from langchain.chat_models import ChatOpenAI
@@ -46,7 +46,7 @@ def process_file(file):
46
 
47
  embeddings = OpenAIEmbeddings()
48
 
49
- pdfsearch = Chroma.from_documents(documents, embeddings)
50
 
51
  chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0.3),
52
  retriever=pdfsearch.as_retriever(search_kwargs={"k": 1}),
 
1
  import gradio as gr
2
  from langchain.embeddings.openai import OpenAIEmbeddings
3
  from langchain.text_splitter import CharacterTextSplitter
4
+ from langchain.vectorstores import Chroma, FAISS
5
 
6
  from langchain.chains import ConversationalRetrievalChain
7
  from langchain.chat_models import ChatOpenAI
 
46
 
47
  embeddings = OpenAIEmbeddings()
48
 
49
+ pdfsearch = FAISS.from_documents(documents, embeddings)
50
 
51
  chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0.3),
52
  retriever=pdfsearch.as_retriever(search_kwargs={"k": 1}),