21-12-13 vernew2

app.py

@@ -7,16 +7,23 @@ import streamlit as st
 import torch 
 #import transformers 
 from transformers import pipeline
+from PIL import Image
+
+
+
+image = Image.open('hb.jpg')
+
 #from transformers import 
 st.header("KNU- Abstractive Summarizer Machine!")
+st.image(image, caption='Welcome to KNU Summarizer')
 
 
 
 
 ''' 
 T5: 너무 오래걸린다.
-
-
+BART: 60 sec 정도
+Pegasus: 
 
 '''
 plms =["facebook/bart-large-cnn", "google/pegasus-xsum", "t5-base" ]
@@ -80,7 +87,7 @@ if submit_button:
     st.success("Congrats!!! ALL DONE!")
     st.balloons()
     
-    balloon_button = st.form_submit_button(label='More Balloon?')
+    balloon_button = st.button(label='More Balloon?')
 
     if balloon_button:    
         st.balloons()