Update src/model_load.py

src/model_load.py

@@ -47,7 +47,7 @@ def load_model():
     # Crear una cadena de LLMChain que combine el retriever y el prompt
     qa_chain = LLMChain(llm=llm, prompt=prompt)
     return qa_chain
-def ask(pregunta: str,retriever):
+def ask(pregunta: str,retriever,qa_chain):
 
     #Busqueda de documentos mediante el retriever
     documentos=retriever.get_relevant_documents(pregunta)