Update functions.py
Browse files- functions.py +1 -2
functions.py
CHANGED
@@ -352,8 +352,7 @@ def chunk_and_preprocess_text(text,thresh=500):
|
|
352 |
def summarize_text(text_to_summarize,max_len,min_len):
|
353 |
'''Summarize text with HF model'''
|
354 |
|
355 |
-
summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,
|
356 |
-
encoder_no_repeat_ngram_size=3,
|
357 |
repetition_penalty=3.5,
|
358 |
num_beams=4,
|
359 |
early_stopping=True)
|
|
|
352 |
def summarize_text(text_to_summarize,max_len,min_len):
|
353 |
'''Summarize text with HF model'''
|
354 |
|
355 |
+
summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,
|
|
|
356 |
repetition_penalty=3.5,
|
357 |
num_beams=4,
|
358 |
early_stopping=True)
|