Update app.py

app.py

@@ -2,7 +2,7 @@ from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
 from datasets import load_dataset
 
 # Load the dataset - Here we use the wiki_dpr dataset for retrieval
-dataset = load_dataset('wiki_dpr', 'psgs_w100.nq.exact')
+dataset = load_dataset('wiki_dpr')
 
 # Initialize the RAG tokenizer (use the T5 tokenizer for RAG)
 tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")