spam_classifier / inference.py
ncfad's picture
Upload 6 files
61c5e06 verified
Raw History Blame Contribute Delete
220 Bytes
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"