Update model.py

model.py

@@ -35,8 +35,13 @@ def run():
     st.write('You entered:', user_input)
 
     # button
-    if st.button(label='Predict'):
+    if st.button(label='Identify Your Feelings'):
         X= preprocess_text(user_input)
         result= prediction([X])
         st.write(result)
-
+        if emo == 0:
+            st.write("Don't worry, you'll be fine")
+        elif emo == 1:
+            st.write('Calm down bro')
+        else:
+            st.write("That's Great!")