Update app.py
Browse files
app.py
CHANGED
@@ -110,11 +110,14 @@ def get_pdf_index():
|
|
110 |
for doc in documents:
|
111 |
texts.extend(splitter.split_text(doc.page_content))
|
112 |
|
|
|
|
|
|
|
113 |
vectorstore_index_creator = VectorstoreIndexCreator(
|
114 |
-
vectorstore_cls=FAISS,
|
115 |
-
embedding_function=
|
116 |
)
|
117 |
-
|
118 |
index = vectorstore_index_creator.from_documents([Document(page_content=text) for text in texts])
|
119 |
|
120 |
return index
|
|
|
110 |
for doc in documents:
|
111 |
texts.extend(splitter.split_text(doc.page_content))
|
112 |
|
113 |
+
|
114 |
+
embedding_function = SentenceTransformer("togethercomputer/m2-bert-80M-8k-retrieval", trust_remote_code=True)
|
115 |
+
|
116 |
vectorstore_index_creator = VectorstoreIndexCreator(
|
117 |
+
vectorstore_cls=FAISS,
|
118 |
+
embedding_function=embedding_function
|
119 |
)
|
120 |
+
|
121 |
index = vectorstore_index_creator.from_documents([Document(page_content=text) for text in texts])
|
122 |
|
123 |
return index
|