abdull4h commited on
Commit
f609ae7
·
verified ·
1 Parent(s): a8285e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -109,7 +109,12 @@ class Vision2030Service:
109
  embedding_function = HuggingFaceEmbeddings(
110
  model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
111
  )
112
- self.vector_store = FAISS.load_local(vector_store_dir, embedding_function)
 
 
 
 
 
113
  else:
114
  print("Creating new vector store...")
115
  # Process PDFs
@@ -447,6 +452,12 @@ def main():
447
 
448
  questions_md = "\n".join([f"- {q}" for q in sample_questions])
449
  gr.Markdown(questions_md)
 
 
 
 
 
 
450
 
451
  return demo
452
 
 
109
  embedding_function = HuggingFaceEmbeddings(
110
  model_name="sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
111
  )
112
+ # Important: Add allow_dangerous_deserialization=True to fix the pickle error
113
+ self.vector_store = FAISS.load_local(
114
+ vector_store_dir,
115
+ embedding_function,
116
+ allow_dangerous_deserialization=True # Add this parameter
117
+ )
118
  else:
119
  print("Creating new vector store...")
120
  # Process PDFs
 
452
 
453
  questions_md = "\n".join([f"- {q}" for q in sample_questions])
454
  gr.Markdown(questions_md)
455
+
456
+ # Add a button to auto-initialize the system when viewing sample questions
457
+ auto_init_btn = gr.Button("Initialize System First")
458
+ auto_init_status = gr.Textbox(label="Initialization Status")
459
+
460
+ auto_init_btn.click(initialize_system, None, auto_init_status)
461
 
462
  return demo
463