Update app.py
Browse files
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 = "./
|
131 |
-
index_path = "./
|
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",
|