Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
from langchain.document_loaders import PyPDFLoader
|
| 3 |
from langchain.text_splitter import CharacterTextSplitter
|
| 4 |
from langchain.embeddings import SentenceTransformerEmbeddings
|
|
|
|
| 5 |
from langchain.vectorstores import FAISS
|
| 6 |
from langchain.chains import RetrievalQA
|
| 7 |
from langchain.llms import HuggingFacePipeline
|
|
@@ -36,7 +37,7 @@ if uploaded_file:
|
|
| 36 |
docs = splitter.split_documents(documents)
|
| 37 |
|
| 38 |
# Embed & Store
|
| 39 |
-
embedding =
|
| 40 |
db = FAISS.from_documents(docs, embedding)
|
| 41 |
retriever = db.as_retriever()
|
| 42 |
|
|
|
|
| 2 |
from langchain.document_loaders import PyPDFLoader
|
| 3 |
from langchain.text_splitter import CharacterTextSplitter
|
| 4 |
from langchain.embeddings import SentenceTransformerEmbeddings
|
| 5 |
+
from langchain_huggingface import HuggingFaceEmbeddings
|
| 6 |
from langchain.vectorstores import FAISS
|
| 7 |
from langchain.chains import RetrievalQA
|
| 8 |
from langchain.llms import HuggingFacePipeline
|
|
|
|
| 37 |
docs = splitter.split_documents(documents)
|
| 38 |
|
| 39 |
# Embed & Store
|
| 40 |
+
embedding = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
| 41 |
db = FAISS.from_documents(docs, embedding)
|
| 42 |
retriever = db.as_retriever()
|
| 43 |
|