enpaiva commited on
Commit
4c798e0
·
verified ·
1 Parent(s): 9fc6315

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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.from_index(DOC_PATH, n_gpu=None)
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")