Update app.py

app.py

@@ -29,9 +29,9 @@ def extract_text_from_pdf(pdf_file):
                 text += page_text + "\n"
     return text
 
-# Load dataset (wiki_dpr) and set trust_remote_code=True
-def load_wiki_dpr():
-    return load_dataset('wiki_dpr', trust_remote_code=True)
+# Load dataset (using SQuAD v2 as a placeholder)
+def load_squad_v2():
+    return load_dataset('squad_v2')
 
 # Store the PDF text and embeddings
 pdf_text = ""
@@ -58,6 +58,9 @@ if pdf_file:
     else:
         st.warning("No extractable text found in the PDF.")
 
+# Load the SQuAD v2 dataset as an example for RAG retrieval
+dataset = load_squad_v2()
+
 # User input for chatbot
 user_input = st.text_input("Ask a question related to the PDF or CSV:")
 
@@ -81,3 +84,4 @@ if st.button("Get Response"):
         except Exception as e:
             st.error(f"Error while processing the question: {e}")
 
+