pmelnechuk commited on
Commit
3571f6c
·
verified ·
1 Parent(s): 48b1569

prueba runnable sequence

Browse files
Files changed (1) hide show
  1. src/model_load.py +1 -1
src/model_load.py CHANGED
@@ -45,7 +45,7 @@ def load_model():
45
  prompt = PromptTemplate(input_variables=["documento", "pregunta"], template=prompt_template)
46
 
47
  # Crear una cadena de LLMChain que combine el retriever y el prompt
48
- qa_chain = LLMChain(llm=llm, prompt=prompt)
49
  return qa_chain
50
  def ask(pregunta: str,retriever,qa_chain):
51
 
 
45
  prompt = PromptTemplate(input_variables=["documento", "pregunta"], template=prompt_template)
46
 
47
  # Crear una cadena de LLMChain que combine el retriever y el prompt
48
+ qa_chain = prompt | llm
49
  return qa_chain
50
  def ask(pregunta: str,retriever,qa_chain):
51