File size: 220 Bytes
61c5e06 | 1 2 3 4 5 6 7 8 9 10 | 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"
|