uyen13 commited on
Commit
cdfff89
·
verified ·
1 Parent(s): 18e02d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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 = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
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