Baweja commited on
Commit
7e5372e
1 Parent(s): 2934f0a

Update app_R.py

Browse files
Files changed (1) hide show
  1. app_R.py +2 -2
app_R.py CHANGED
@@ -5,8 +5,8 @@ from transformers import RagRetriever, RagSequenceForGeneration
5
  # Load model
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
- dataset_path = "./sample/my_knowledge_dataset"
9
- index_path = "./sample/my_knowledge_dataset_hnsw_index.faiss"
10
 
11
  retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name="custom",
12
  passages_path = dataset_path,
 
5
  # Load model
6
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
+ dataset_path = "./5K_index_data/my_knowledge_dataset"
9
+ index_path = "./5K_index_data/my_knowledge_dataset_hnsw_index.faiss"
10
 
11
  retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name="custom",
12
  passages_path = dataset_path,