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)
    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')