Chan-Y commited on
Commit
65033bf
1 Parent(s): 367b557

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
app.py CHANGED
@@ -67,10 +67,16 @@ The summary should be in {LANG} language.
67
  template=template_translate,
68
  input_variables=["TEXT", "LANG"]
69
  )
70
- formatted_prompt = prompt_summarize.format(TEXT=text, LANG=lang)
71
- summary = llm.invoke(formatted_prompt)
72
 
73
- return summary
 
 
 
 
 
 
 
 
74
 
75
  def download_summary(output_text):
76
  if output_text:
 
67
  template=template_translate,
68
  input_variables=["TEXT", "LANG"]
69
  )
 
 
70
 
71
+ summaries = []
72
+ for i in range(0, len(text), CONTEXT_WINDOW):
73
+ chunk = text[i:i + CONTEXT_WINDOW]
74
+ formatted_prompt = prompt_summarize.format(TEXT=chunk, LANG=lang)
75
+ summary = llm.invoke(formatted_prompt)
76
+ summaries.append(summary)
77
+
78
+ final_summary = "\n\n".join(summaries)
79
+ return final_summary
80
 
81
  def download_summary(output_text):
82
  if output_text: