Update app.py

app.py

@@ -17,14 +17,15 @@ model = load_model()
 #prompts
 st.title("Writing Assistant for you 🦄")
 
-textbox = st.text_area('Write your text:', '', height=200, max_chars=1000)
+textbox = st.text_area('Write your text in this box:', '', height=200, max_chars=1000)
 
 button = st.button('Detect grammar mistakes:')
 
 if button:
-    output_text = model(textbox)[0]
-    st.markdown(output_text)
-
+    output_text = model(textbox)[0]["generated_text]
+    
+    st.write("Correct : ", output_text)
+    
 #inputs = tokenizer("Grammar: "+sent,return_tensors="tf")
 
 #output_sequences = infer(inputs)