Update app.py

app.py

@@ -59,8 +59,7 @@ def predict(new_sentence):
 if text or aButton:
     with col2:
         with st.spinner('Wait for it...'):
-            st.header(predict(text))
-            st.success("Prediction Successful!")
+            st.success(predict(text))