Update app.py

app.py

@@ -2,11 +2,11 @@ import streamlit as st
 from transformers import pipeline
 
 @st.cache(allow_output_mutation=True)
-def load_summarizer():
-    model = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
-    return model
+#def load_summarizer():
+model = pipeline("summarization", model="google/bigbird-pegasus-large-bigpatent")
+    #return model
 
-summarizer = load_summarizer()
+#summarizer = load_summarizer()
 st.title("Patent Text Summarizer")
 sentence = st.text_area('Please paste your Patent Text :', height=300)
 button = st.button("Summarize")
@@ -17,9 +17,10 @@ button = st.button("Summarize")
 with st.spinner("Generating Patent Summary.."):
     if button and sentence:
         #chunks = generate_chunks(sentence)
-        res = summarizer(sentence,
+        res = model(sentence,
                          max_length=500, 
-                         min_length=100, 
+                         min_length=100,
+                         truncation=True,
                          #do_sample=do_sample
                          )
         text = ' '.join([summ['summary_text'] for summ in res])