Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,8 @@ def preprocess_pdfs(folder_path, save_vectorstore_path):
|
|
| 41 |
@st.cache_resource
|
| 42 |
def load_vector_store(save_vectorstore_path):
|
| 43 |
embedding_function = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
| 44 |
-
return FAISS.load_local(save_vectorstore_path, embedding_function)
|
|
|
|
| 45 |
|
| 46 |
|
| 47 |
# Generate summary based on the retrieved text
|
|
@@ -84,4 +85,3 @@ if __name__ == "__main__":
|
|
| 84 |
main()
|
| 85 |
|
| 86 |
|
| 87 |
-
|
|
|
|
| 41 |
@st.cache_resource
|
| 42 |
def load_vector_store(save_vectorstore_path):
|
| 43 |
embedding_function = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
|
| 44 |
+
return FAISS.load_local(save_vectorstore_path, embedding_function, allow_dangerous_deserialization=True)
|
| 45 |
+
|
| 46 |
|
| 47 |
|
| 48 |
# Generate summary based on the retrieved text
|
|
|
|
| 85 |
main()
|
| 86 |
|
| 87 |
|
|
|