Spaces:
Runtime error
Runtime error
File size: 372 Bytes
ec159e6 | 1 2 3 4 5 6 7 8 9 10 11 | 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
|