Update app.py

app.py

@@ -17,7 +17,7 @@ model = load_model()
     
 #prompts
 st.title("Writing Assistant for you 🦄")
-image = Image.open('grammar.jpg').resize((1000,500))
+image = Image.open('grammar.jpg').resize((1000,300))
 st.image(image, caption='Sunrise by the mountains')
 st.subheader("Some examples: ")
 example_1 = st.button("I am write on AI")
@@ -31,11 +31,13 @@ button = st.button('Detect grammar mistakes:')
 if example_1:
     with st.spinner('In progress.......'):
         output_text = model("I am write on AI")[0]["generated_text"]
-    st.markdown("**"+output_text+"**")
+    st.markdown("## "+output_text)
 
 if example_2:
-    output_text = model("This sentence has, bads grammar mistake!")[0]["generated_text"]
+    with st.spinner('In progress.......'):
+        output_text = model("This sentence has, bads grammar mistake!")[0]["generated_text"]
     st.markdown("**"+output_text+"**")
 if button:
-    output_text = model(textbox)[0]["generated_text"]
+    with st.spinner('In progress.......'):
+        output_text = model(textbox)[0]["generated_text"]
     st.markdown("**"+output_text+"**")