Spaces:
Build error
Build error
| 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 | |