Update functions.py

functions.py

@@ -84,6 +84,19 @@ def sentiment_pipe(earnings_text):
     
     return earnings_sentiment, earnings_sentences    
 
+@st.experimental_memo(suppress_st_warning=True)
+def summarize_text(text_to_summarize,max_len,min_len):
+    '''Summarize text with HF model'''
+    
+    summarized_text = sum_pipe(text_to_summarize,max_length=max_len,min_length=min_len,clean_up_tokenization_spaces=True,no_repeat_ngram_size=4,
+           encoder_no_repeat_ngram_size=3,
+           repetition_penalty=3.5,
+           num_beams=4,
+           early_stopping=True)
+     summarized_text = ' '.join([summ['summary_text'] for summ in summarized_text])
+     
+     return summarized_text
+     
 @st.experimental_memo(suppress_st_warning=True)
 def clean_text(text):
     '''Clean all text'''