ShubhamAC's picture
Initial commit: ML inference and evaluation pipeline
0b01ce8
raw
history blame contribute delete
454 Bytes
import joblib
import os
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
MODEL_DIR = os.path.join(BASE_DIR, "models")
vectorizer = None
model = None
def load_model():
global vectorizer, model
if vectorizer is None or model is None:
vectorizer = joblib.load(os.path.join(MODEL_DIR, "tfidf_vectorizer.pkl"))
model = joblib.load(os.path.join(MODEL_DIR, "logistic_model.pkl"))
return vectorizer, model