ver-new-1

app.py

@@ -10,14 +10,19 @@ from transformers import pipeline
 #from transformers import 
 st.header("KNU- Abstractive Summarizer Machine!")
 
-st.write("Information")
-st.write("__Inputs__: Text your input article!!")
-st.write("__Outputs__: Summarizing output text by Google-Pegasus! ")
 
 
+#t5: 너무 오래걸린다.
+
 plms =["facebook/bart-large-cnn", "google/pegasus-xsum", "t5-base" ]
 
 
+st.write("Information")
+st.write("__Inputs__: Text your input article!!")
+st.write("__Outputs__: Summarizing output text by State-of-the-art NLP summarization Models! ")
+
+
+
 def load_plms(model_name):
     #model_name = "google/pegasus-xsum"
     summarizer = pipeline(task="summarization", model=model_name) 
@@ -56,13 +61,13 @@ output_text = []
 if submit_button:
     with st.spinner('On summarizing !...wait a second please..'):
         output_text.append(get_summarizer(summarizer_1, text_input, 150, 5))
-        output_text.append(get_summarizer(summarizer_2, text_input, 150, 5))
-        output_text.append(get_summarizer(summarizer_3, text_input, 150, 5))
+        #output_text.append(get_summarizer(summarizer_2, text_input, 150, 5))
+        #output_text.append(get_summarizer(summarizer_3, text_input, 150, 5))
    
     
     st.markdown("### Outputs are here !:  ")
     
-    for i in range(3):
+    for i in range(1):
         st.markdown("**"+ plms[i] +"s Output:  **  ")
         st.text(output_text[i])
         st.success(f"{i+1} of 3 are done!")