import joblib

model = joblib.load("model.pkl")
tfidf = joblib.load("vectorizer.pkl")

def predict(text):
    X = tfidf.transform([text])
    pred = model.predict(X)[0]
    return "spam" if pred == 1 else "ham"