Spaces:

arifa-batool
/

spam-filter-app

Runtime error

spam-filter-app / utils /model_loader.py

Email Spam Classifier using ML and UI in Gradio with Jupyter Notebook

ec159e6 verified 10 months ago

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

	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