Update app.py

app.py

@@ -1,8 +1,8 @@
 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')
+# Load the wiki_dpr dataset and trust the remote code to execute
+dataset = load_dataset("wiki_dpr", trust_remote_code=True)
 
 # Initialize the RAG tokenizer (use the T5 tokenizer for RAG)
 tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")