Spaces:
Runtime error
Runtime error
File size: 546 Bytes
6cf89af |
1 2 3 4 5 6 7 8 9 10 11 12 |
import os
PROJECT_DIR = os.path.dirname(os.path.abspath(__file__))
MODELS_DIR = os.path.join(PROJECT_DIR, 'models')
FLAIR_MODEL_PATH = os.path.join(MODELS_DIR, 'flair-sentiment-classifier/best-model.pt')
SPACY_MODEL_PATH = os.path.join(MODELS_DIR, 'spacy-classifier/model-best')
SETFIT_MODEL_PATH = os.path.join(MODELS_DIR, 'setfit-classifier')
NB_MODEL_PATH = os.path.join(MODELS_DIR, 'm_nb.sav')
COUNT_VECTORIZER_PATH = os.path.join(MODELS_DIR, 'count_vectorizer.sav')
TFIDF_VECTORIZER_PATH = os.path.join(MODELS_DIR, 'tfidf_vectorizer.sav') |