Update app.py

app.py

@@ -33,13 +33,13 @@ if submit:
     num_return_sequences=3
   )
   
+  st.subheader("Most probables: ")
   for y in outputs:
-    st.subheader("Most probable: ")
     
     out_text = ttokenizer.decode(y, skip_special_tokens=True)
     st.success(out_text.capitalize())
     
-    st.markdown("### Edited sentences:")
+    st.markdown("#### Highlighter on edits:")
     c_text = ""
     for x in out_text.lower().split(" "):
       if x in input_text.lower().split(" "):