Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,9 @@ def preprocess_pdfs(folder_path, save_vectorstore_path):
|
|
| 40 |
# Load pre-trained FAISS vector store
|
| 41 |
@st.cache_resource
|
| 42 |
def load_vector_store(save_vectorstore_path):
|
| 43 |
-
|
|
|
|
|
|
|
| 44 |
|
| 45 |
# Generate summary based on the retrieved text
|
| 46 |
def generate_summary_with_huggingface(query, retrieved_text):
|
|
@@ -81,3 +83,4 @@ def main():
|
|
| 81 |
if __name__ == "__main__":
|
| 82 |
main()
|
| 83 |
|
|
|
|
|
|
| 40 |
# Load pre-trained FAISS vector store
|
| 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
|
| 48 |
def generate_summary_with_huggingface(query, retrieved_text):
|
|
|
|
| 83 |
if __name__ == "__main__":
|
| 84 |
main()
|
| 85 |
|
| 86 |
+
|