Update app.py

app.py

@@ -19,7 +19,8 @@ def question_model():
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def summarization_model():
-    summarizer = pipeline("summarization")
+    model_name = "google/pegasus-xsum"
+    summarizer = pipeline(model=model_name, tokenizer=model_name, task="summarization")
     return summarizer
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
@@ -62,17 +63,18 @@ if option == "Extractive question answering":
                     answer = answer["answer"]
                     st.text(answer)
 
-elif option == 'Text summarization':
+elif option == "Text summarization":
     st.markdown("<h2 style='text-align: center; color:red;'>Summarize text</h2>", unsafe_allow_html=True)
-    sample_text = "sample text"
     source = st.radio("How would you like to start? Choose an option below", ["I want to input some text", "I want to upload a file"])
     if source == "I want to input some text":
-        text = st.text_area('Input a text in English (between 1,000 and 10,000 characters)', value=sample_text, max_chars=10000, height=330)
-        button = st.button('Get summary')
+        with open("sample.txt", "r") as text_file:
+            sample_text = text_file.read()
+        text = st.text_area("Input a text in English (10,000 characters max)", value=sample_text, max_chars=10000, height=330)
+        button = st.button("Get summary")
         if button:
             summarizer = summarization_model()
             with st.spinner(text="Summarizing text..."):
-                summary = summarizer(text, max_length=130, min_length=30)
+                summary = summarizer(text)
                 st.write(summary)
 
     elif source == "I want to upload a file":