Update app.py

app.py

@@ -11,6 +11,7 @@ retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name=
 model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq", retriever=retriever)
 
 # Load the wiki_dpr dataset with trust_remote_code=True
+# Ensure to replace with the correct dataset name
 dataset = load_dataset("facebook/wiki_dpr", split="train", trust_remote_code=True)
 
 # Function to get RAG embeddings