dron3flyv3r commited on
Commit
a1d7b67
·
1 Parent(s): 8d92114

Update summarize_text function to print summary before returning

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,7 +20,8 @@ def summarize_text(text: str, bullet_points: int, conclusion: bool) -> str:
20
  else:
21
  prompt = f"Summarize the following text into {bullet_points} bullet points: {text}"
22
  summary = api.text_generation(prompt, max_new_tokens=250, do_sample=True)
23
- return summary["generated_text"]
 
24
 
25
  def control(audio_file, text: str, bullet_points: int, conclusion: bool) -> str:
26
  if audio_file:
 
20
  else:
21
  prompt = f"Summarize the following text into {bullet_points} bullet points: {text}"
22
  summary = api.text_generation(prompt, max_new_tokens=250, do_sample=True)
23
+ print(summary)
24
+ return summary
25
 
26
  def control(audio_file, text: str, bullet_points: int, conclusion: bool) -> str:
27
  if audio_file: