Update app.py

app.py

@@ -1,12 +1,36 @@
 import streamlit as st
 from happytransformer import HappyTextToText, TTSettings
 
-happy_tt = HappyTextToText("T5", "team-writing-assistant/t5-base-c4jfleg")
+checkpoint = "team-writing-assistant/t5-base-c4jfleg"
+happy_tt = HappyTextToText("T5", checkpoint)
 args = TTSettings(num_beams=5, min_length=1)
 
-input = st.text_area('Enter some text')
+st.title("Check & Improve English Grammar")
+st.markdown("This writing assistant detects 🔍 and corrects ✍️ grammatical mistakes for you!")
 
-if input:
-  input = "grammar: " + input
-  result = happy_tt.generate_text(input, args=args)
-  st.text(result.text)
+st.subheader("Example text: ")
+col1, col2 = st.columns([1,1])
+with col1:
+  example_1 = st.button("Speed of light is fastest then speed of sound")
+with col2:
+  example_2 = st.button("Who are the president?")
+
+input_text = st.text_area('Enter your text here')
+button = st.button('Submit')
+
+def output(input_text):
+  with st.spinner('Detecting 🔍..'):
+    input_text = "grammar: " + input_text
+    result = happy_tt.generate_text(input_text, args=args)
+    st.markdown("## " + result.text)
+
+if example_1:
+  output("Speed of light is fastest then speed of sound")
+if example_2:
+  output("Who are the president?")  
+if input_text:
+  output(input_text)
+  
+st.subheader("")
+
+st.text("Built by Team Writing Assistant ❤️")