richardorama
commited on
Commit
•
954b204
1
Parent(s):
6d6b1c1
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
|
|
9 |
import ast
|
10 |
|
11 |
#st.title("Assorted Language Tools")
|
12 |
-
st.markdown("<h1 style='text-align: center; font-size:
|
13 |
st.markdown("<h3 style='text-align: center; font-size: 16px;'>Simply Assorted Language Tools</h3>", unsafe_allow_html=True)
|
14 |
st.markdown("<h3 style='text-align: center; font-size: 20px; color: blue;'>Orama's AI Craze</h3>", unsafe_allow_html=True)
|
15 |
|
@@ -26,7 +26,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
|
26 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
27 |
|
28 |
# Streamlit UI for input
|
29 |
-
st.markdown("<h3 style='text-align: center; font-size: 20px; background-color:
|
30 |
|
31 |
# Input text area for the article
|
32 |
article = st.text_area("Enter the text you want to summarize", height=300)
|
@@ -212,7 +212,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
|
|
212 |
gpt_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
213 |
|
214 |
# Streamlit UI
|
215 |
-
st.markdown("<h3 style='text-align: center; font-size: 20px; background-color:
|
216 |
|
217 |
if 'conversation' not in st.session_state:
|
218 |
st.session_state.conversation = ""
|
|
|
9 |
import ast
|
10 |
|
11 |
#st.title("Assorted Language Tools")
|
12 |
+
st.markdown("<h1 style='text-align: center; font-size: 30px;'>S A L T</h1>", unsafe_allow_html=True)
|
13 |
st.markdown("<h3 style='text-align: center; font-size: 16px;'>Simply Assorted Language Tools</h3>", unsafe_allow_html=True)
|
14 |
st.markdown("<h3 style='text-align: center; font-size: 20px; color: blue;'>Orama's AI Craze</h3>", unsafe_allow_html=True)
|
15 |
|
|
|
26 |
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
|
27 |
|
28 |
# Streamlit UI for input
|
29 |
+
st.markdown("<h3 style='text-align: center; font-size: 20px; background-color: white; color: black;'>Text Summarization with BART</h3>", unsafe_allow_html=True)
|
30 |
|
31 |
# Input text area for the article
|
32 |
article = st.text_area("Enter the text you want to summarize", height=300)
|
|
|
212 |
gpt_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
213 |
|
214 |
# Streamlit UI
|
215 |
+
st.markdown("<h3 style='text-align: center; font-size: 20px; background-color: white; color: black;'>Chat with GPT</h3>", unsafe_allow_html=True)
|
216 |
|
217 |
if 'conversation' not in st.session_state:
|
218 |
st.session_state.conversation = ""
|