Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import RagRetriever, RagTokenizer, RagSequenceForGeneration
|
|
3 |
from datasets import load_dataset
|
4 |
|
5 |
# Step 1: Load the dataset with the trust_remote_code flag enabled and a valid config name
|
6 |
-
dataset = load_dataset("wiki_dpr", "
|
7 |
|
8 |
# Step 2: Load the retriever using the pre-trained model, with use_dummy_dataset=True and trust_remote_code=True
|
9 |
retriever = RagRetriever.from_pretrained(
|
|
|
3 |
from datasets import load_dataset
|
4 |
|
5 |
# Step 1: Load the dataset with the trust_remote_code flag enabled and a valid config name
|
6 |
+
dataset = load_dataset("wiki_dpr", "psgs_w100.nq.exact", trust_remote_code=True)
|
7 |
|
8 |
# Step 2: Load the retriever using the pre-trained model, with use_dummy_dataset=True and trust_remote_code=True
|
9 |
retriever = RagRetriever.from_pretrained(
|