spam-filter-app / utils /model_loader.py
arifa-batool's picture
Email Spam Classifier using ML and UI in Gradio with Jupyter Notebook
ec159e6 verified
raw
history blame contribute delete
372 Bytes
import pickle
def load_models(vectorizer_path="saved_models/vectorizer_TF-IDF.pkl",
model_path="saved_models/SVM_TF-IDF.pkl"):
"""Load vectorizer and SVM model."""
with open(vectorizer_path, "rb") as f:
vectorizer = pickle.load(f)
with open(model_path, "rb") as f:
model = pickle.load(f)
return vectorizer, model