pmelnechuk commited on
Commit
3e6183d
·
verified ·
1 Parent(s): c8a9ecb

Update src/model_load.py

Browse files
Files changed (1) hide show
  1. src/model_load.py +1 -1
src/model_load.py CHANGED
@@ -50,7 +50,7 @@ def load_model():
50
  def ask(pregunta: str,retriever,qa_chain):
51
 
52
  #Busqueda de documentos mediante el retriever
53
- documentos=retriever.get_relevant_documents(pregunta)
54
 
55
  #Generacion de la respuesta
56
  respuesta = qa_chain.invoke({
 
50
  def ask(pregunta: str,retriever,qa_chain):
51
 
52
  #Busqueda de documentos mediante el retriever
53
+ documentos=retriever.invoke(pregunta)
54
 
55
  #Generacion de la respuesta
56
  respuesta = qa_chain.invoke({