Update app.py
Browse files
app.py
CHANGED
|
@@ -57,6 +57,7 @@ MODEL_NAME = "facebook/bart-large-cnn" # A commonly used summarization model
|
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 58 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
| 59 |
|
|
|
|
| 60 |
DEFAULT_STATEMENT = ""
|
| 61 |
# Create a text area for user input
|
| 62 |
STATEMENT = st.sidebar.text_area('Enter Statement (String1)', DEFAULT_STATEMENT, height=150)
|
|
|
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 58 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
| 59 |
|
| 60 |
+
st.sidebar.markdown("<h3 style='text-align: center; font-size: 20px; background-color: white; color: black;'>xxxText Summarization with BART</h3>", unsafe_allow_html=True)
|
| 61 |
DEFAULT_STATEMENT = ""
|
| 62 |
# Create a text area for user input
|
| 63 |
STATEMENT = st.sidebar.text_area('Enter Statement (String1)', DEFAULT_STATEMENT, height=150)
|