Update app.py

app.py

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