richardorama commited on
Commit
954b204
1 Parent(s): 6d6b1c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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: 20px;'>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,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: yellow;'>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,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: yellow;'>Chat with GPT</h3>", unsafe_allow_html=True)
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 = ""