ZeeAI1 commited on
Commit
58ab5fb
·
verified ·
1 Parent(s): 1a235fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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
- return FAISS.load_local(save_vectorstore_path, embedding_function=HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2"))
 
 
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
+