Update functions.py

functions.py

@@ -323,7 +323,7 @@ def sentiment_pipe(earnings_text):
     return earnings_sentiment, earnings_sentences 
 
 @st.cache_data
-def chunk_and_preprocess_text(text, model_name):
+def chunk_and_preprocess_text(text, model_name= 'philschmid/flan-t5-base-samsum'):
 
     '''Chunk and preprocess text for summarization'''
     
@@ -341,22 +341,22 @@ def chunk_and_preprocess_text(text, model_name):
     combined_length = len(tokenizer.tokenize(sentence)) + length # add the no. of sentence tokens to the length counter
     
     if combined_length  <= tokenizer.max_len_single_sentence: # if it doesn't exceed
-      chunk += sentence + " " # add the sentence to the chunk
-      length = combined_length # update the length counter
+        chunk += sentence + " " # add the sentence to the chunk
+        length = combined_length # update the length counter
     
       # if it is the last sentence
-      if count == len(sentences) - 1:
-        chunks.append(chunk) # save the chunk
+        if count == len(sentences) - 1:
+            chunks.append(chunk) # save the chunk
       
     else: 
-      chunks.append(chunk) # save the chunk
-      # reset 
-      length = 0 
-      chunk = ""
-    
-      # take care of the overflow sentence
-      chunk += sentence + " "
-      length = len(tokenizer.tokenize(sentence))
+        chunks.append(chunk) # save the chunk
+        # reset 
+        length = 0 
+        chunk = ""
+        
+        # take care of the overflow sentence
+        chunk += sentence + " "
+        length = len(tokenizer.tokenize(sentence))
 
     return chunks