Update app.py

app.py

@@ -2,11 +2,16 @@ from joblib import load
 from sklearn.feature_extraction.text import TfidfVectorizer
 import numpy as np
 import streamlit as st
-
+st.header('Spam Detection using Naive Bayes Classifier')
 vectorizer = load('tfidf_vectorizer.joblib')
 user_input = st.text_input("Enter some text:", "")
 if user_input is not None:
     x = vectorizer.transform([user_input])
     model = load('Naive_Bayes_Spam_Detection.joblib')
     pred = model.predict(x)
-    st.write(pred)
+    if pred[0] == 1:
+        st.write('The entered text is a Spam')
+    elif pred[0] == 0:
+        st.write('The entered text is not a Spam')
+    else:
+        st.write('Error, Try again')