Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
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 |
|