Shankarm08 commited on
Commit
b17e18d
·
verified ·
1 Parent(s): 46635b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -11,7 +11,6 @@ retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name=
11
  model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq", retriever=retriever)
12
 
13
  # Load the wiki_dpr dataset with trust_remote_code=True
14
- # Ensure to replace with the correct dataset name
15
  dataset = load_dataset("facebook/wiki_dpr", split="train", trust_remote_code=True)
16
 
17
  # Function to get RAG embeddings
 
11
  model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq", retriever=retriever)
12
 
13
  # Load the wiki_dpr dataset with trust_remote_code=True
 
14
  dataset = load_dataset("facebook/wiki_dpr", split="train", trust_remote_code=True)
15
 
16
  # Function to get RAG embeddings