Update app.py

app.py

@@ -2,7 +2,7 @@ from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
 from datasets import load_dataset
 
 # Load the wiki_dpr dataset and trust the remote code to execute
-dataset = load_dataset("wiki_dpr", trust_remote_code=True)
+dataset = load_dataset('wiki_dpr', 'psgs_w100.nq.exact')
 
 # Initialize the RAG tokenizer (use the T5 tokenizer for RAG)
 tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")