Update model.py

model.py

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