Update app.py

app.py

@@ -18,6 +18,7 @@ import torch
 from Text_analysis import *
 from Metadata import *
 from app_utils import *
+from PIL import Image
 
 
 HTML_BANNER = """
@@ -25,6 +26,9 @@ HTML_BANNER = """
     <h1 style="color:white;text-align:center;">Summary app </h1>
     </div>
     """
+def load_image(file):
+    img = Image.open(file)
+    return img
 
 
 def main():
@@ -34,6 +38,7 @@ def main():
 
     if choice=='Summarization':
         stc.html(HTML_BANNER)
+        st.image(load_image('summary.png'))
         st.subheader('summarization')
         raw_text=st.text_area("Enter the text you want to summarize")
         if st.button("Summarize"):