21-12-13

app.py

@@ -12,8 +12,12 @@ st.header("Header: Summarization")
 st.write("Information")
 st.write("__Inputs__: Text your input article!!")
 st.write("__Outputs__: Summarizing output text by Google-Pegasus! ")
+
+
+model_name= "google/pegasus-xsum"
+
 def load_summary_model():
-    model_name = "google/pegasus-xsum"
+    #model_name = "google/pegasus-xsum"
     summarizer = pipeline(task="summarization", model=model_name) 
     
     return summarizer    
@@ -32,7 +36,7 @@ with st.form(key="my_form"):
     display_text = ARTICLE + "\n\n" 
     text_input = st.text_area("Input any text you want to summaryize & classify here (keep in mind very long text will take a while to process):", display_text)
     submit_button = st.form_submit_button(label='Submit')
-with st.spinner('Loading Pretrained BART Models (_please allow for 10 seconds_)...'):
+with st.spinner('Loading Pretrained', model_name ,'Models (_please allow for 10 seconds_)...'):
     summarizer = load_summary_model()   
     
 if submit_button: