nickmuchi commited on
Commit
59e6849
·
1 Parent(s): bce92b9

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -2
functions.py CHANGED
@@ -364,8 +364,7 @@ def chunk_and_preprocess_text(text, model_name= 'philschmid/flan-t5-base-samsum'
364
  def summarize_text(text_to_summarize,max_len,min_len):
365
  '''Summarize text with HF model'''
366
 
367
- summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,
368
- early_stopping=True)
369
  summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
370
 
371
  return summarized_text
 
364
  def summarize_text(text_to_summarize,max_len,min_len):
365
  '''Summarize text with HF model'''
366
 
367
+ summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,do_sample=False)
 
368
  summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
369
 
370
  return summarized_text