Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -66,9 +66,8 @@ list_nodes = [node.text for node in nodes]
|
|
| 66 |
print(os.getcwd())
|
| 67 |
|
| 68 |
# if os.path.exists(DOC_PATH):
|
| 69 |
-
RAG = RAGPretrainedModel.
|
| 70 |
-
|
| 71 |
-
RAG.model.loaded_from_index = True
|
| 72 |
|
| 73 |
# else:
|
| 74 |
# RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
|
|
|
|
| 66 |
print(os.getcwd())
|
| 67 |
|
| 68 |
# if os.path.exists(DOC_PATH):
|
| 69 |
+
RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
|
| 70 |
+
RAG = RAG.from_index(DOC_PATH, n_gpu=None)
|
|
|
|
| 71 |
|
| 72 |
# else:
|
| 73 |
# RAG = RAGPretrainedModel.from_pretrained("AdrienB134/ColBERTv2.0-spanish-mmarcoES")
|