Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|