sabre-code commited on
Commit
a90190c
1 Parent(s): 6d091d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
-
5
  st.title("Text Summarization App PEGASUS-Large) ")
6
  st.subheader('Input text below to be summarised', divider='rainbow')
7
  # Create a text input widget
@@ -14,7 +14,7 @@ def generate_summary(text):
14
 
15
 
16
  # Return the generated summary
17
- return generated_summary
18
 
19
  # Add a button to trigger the generation of the summary
20
  generate_button = st.button(label="Generate Summary")
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ st,set_page_config(layout='wide')
5
  st.title("Text Summarization App PEGASUS-Large) ")
6
  st.subheader('Input text below to be summarised', divider='rainbow')
7
  # Create a text input widget
 
14
 
15
 
16
  # Return the generated summary
17
+ return generated_summary['summary_text']
18
 
19
  # Add a button to trigger the generation of the summary
20
  generate_button = st.button(label="Generate Summary")