richardorama commited on
Commit
6ba08bc
1 Parent(s): 889b7a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -8,8 +8,8 @@ import streamlit as st
8
  from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
9
  import ast
10
 
11
- st.title("Assorted Language Tools - AI Craze")
12
-
13
 
14
 
15
  ################ STATEMENT SUMMARIZATION - main area #################
@@ -24,7 +24,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
24
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
25
 
26
  # Streamlit UI for input
27
- st.markdown("<h3 style='text-align: center; font-size: 20px;'>Text Summarization with BART</h3>", unsafe_allow_html=True)
28
 
29
  # Input text area for the article
30
  article = st.text_area("Enter the text you want to summarize", height=300)
@@ -210,7 +210,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
210
  gpt_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
211
 
212
  # Streamlit UI
213
- st.markdown("<h3 style='text-align: center; font-size: 20px;'>Chat with GPT</h3>", unsafe_allow_html=True)
214
 
215
  if 'conversation' not in st.session_state:
216
  st.session_state.conversation = ""
 
8
  from transformers import pipeline, GPT2Tokenizer, GPT2LMHeadModel
9
  import ast
10
 
11
+ st.title("Assorted Language Tools")
12
+ st.markdown("<h3 style='text-align: center; font-size: 20px; color> blue;'>Orama's AI Craze</h3>", unsafe_allow_html=True)
13
 
14
 
15
  ################ STATEMENT SUMMARIZATION - main area #################
 
24
  model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
25
 
26
  # Streamlit UI for input
27
+ st.markdown("<h3 style='text-align: center; font-size: 20px; color: red;'>Text Summarization with BART</h3>", unsafe_allow_html=True)
28
 
29
  # Input text area for the article
30
  article = st.text_area("Enter the text you want to summarize", height=300)
 
210
  gpt_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
211
 
212
  # Streamlit UI
213
+ st.markdown("<h3 style='text-align: center; font-size: 20px; color: red;'>Chat with GPT</h3>", unsafe_allow_html=True)
214
 
215
  if 'conversation' not in st.session_state:
216
  st.session_state.conversation = ""