Spaces:
Runtime error
Runtime error
Commit
·
063d1c6
1
Parent(s):
8d40917
Update app.py
Browse files
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 =
|
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}),
|