Baweja commited on
Commit
7cf9753
·
verified ·
1 Parent(s): b0a3c06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -127,8 +127,8 @@ def respond(
127
  # Load model
128
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
129
 
130
- dataset_path = "./IndexedDataFiles/my_knowledge_dataset"
131
- index_path = "./IndexedDataFiles/my_knowledge_dataset_hnsw_index.faiss"
132
 
133
  tokenizer = AutoTokenizer.from_pretrained("facebook/rag-sequence-nq")
134
  retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name="custom",
 
127
  # Load model
128
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
129
 
130
+ dataset_path = "./sample/my_knowledge_dataset"
131
+ index_path = "./sample/my_knowledge_dataset_hnsw_index.faiss"
132
 
133
  tokenizer = AutoTokenizer.from_pretrained("facebook/rag-sequence-nq")
134
  retriever = RagRetriever.from_pretrained("facebook/rag-sequence-nq", index_name="custom",