Update src/model_load.py

src/model_load.py

@@ -49,12 +49,14 @@ def load_model():
     return qa_chain
 def ask(pregunta: str,retriever,qa_chain):
 
+    
+    
     #Busqueda de documentos mediante el retriever
     documentos=retriever.invoke(pregunta)
 
     #Generacion de la respuesta
     respuesta = qa_chain.invoke({
-    "documentos": "\n".join([doc.page_content for doc in documentos]),
+    "documento": "\n".join([doc.page_content for doc in documentos]),
     "pregunta": pregunta
     })