Update functions.py
Browse files- 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,
|
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
|