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",
|