Update app.py

app.py

@@ -9,8 +9,8 @@ Input=st.text_input("Input the message below",)
 if st.button("Check"):
     st.text("Process may take upto a minute. Please be patient. Thank you!")
     df=pd.read_csv("mail_data.csv")
-    df.loc[mail_data['Category'] == 'spam', 'Category'] = 0
-    df.loc[mail_data['Category'] == 'ham', 'Category'] = 1
+    df.loc[df['Category'] == 'spam', 'Category'] = 0
+    df.loc[df['Category'] == 'ham', 'Category'] = 1
     X = df['Message']
     Y = df['Category']
     from keras.utils import pad_sequences