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 |
+
|