Update app.py

app.py

@@ -2,8 +2,8 @@ import torch
 from transformers import RagRetriever, RagTokenizer, RagSequenceForGeneration
 from datasets import load_dataset
 
-# Step 1: Load the dataset with the trust_remote_code flag enabled
-dataset = load_dataset("wiki_dpr", trust_remote_code=True)
+# Step 1: Load the dataset with the trust_remote_code flag enabled and a valid config name
+dataset = load_dataset("wiki_dpr", "psgs_w100.nq.exact", trust_remote_code=True)
 
 # Step 2: Load the retriever using the pre-trained model, with use_dummy_dataset=True and trust_remote_code=True
 retriever = RagRetriever.from_pretrained(