Update app.py

app.py

@@ -19,6 +19,18 @@ st.title("Writing Assistant for you 🦄")
 
 textbox = st.text_area('Write your text:', '', height=200, max_chars=1000)
 
+button = st.button('Detect grammar mistakes:')
+
+if button:
+    output_text = model(textbox)[0]
+ 
+    for i, line in enumerate(output_text.split("\n")):
+        if ":" in line:
+            speaker, speech = line.split(':')
+            st.markdown(f'__{speaker}__: {speech}')
+        else:
+            st.markdown(line)
+
 #inputs = tokenizer("Grammar: "+sent,return_tensors="tf")
 
 #output_sequences = infer(inputs)