NicolasGaudemet commited on
Commit
6dd3f45
1 Parent(s): 0165bf1

Update summarizer_app.py

Browse files
Files changed (1) hide show
  1. summarizer_app.py +1 -1
summarizer_app.py CHANGED
@@ -36,7 +36,7 @@ def summarize_text(text_to_summarize, llm):
36
  SUMMARY:"""
37
  summary_langage_prompt = PromptTemplate(template=prompt_template, input_variables=['text'])
38
  #summary_langage_prompt.format(taille=f"{summary_length}")
39
- chain = load_summarize_chain(llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=summary_langage_prompt)
40
  steps = chain({"input_documents": docs}, return_only_outputs=True)
41
 
42
  print(len(steps['intermediate_steps']))
 
36
  SUMMARY:"""
37
  summary_langage_prompt = PromptTemplate(template=prompt_template, input_variables=['text'])
38
  #summary_langage_prompt.format(taille=f"{summary_length}")
39
+ chain = load_summarize_chain(llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=summary_langage_prompt, combine_prompt = summary_langage_prompt)
40
  steps = chain({"input_documents": docs}, return_only_outputs=True)
41
 
42
  print(len(steps['intermediate_steps']))