dlsmallw's picture
Task-313 Fix regression value output and reformat UI to look better
fa7fec5
raw
history blame
696 Bytes
# Used for setting some constants for the project codebase
from pathlib import Path
# Root Path
ROOT = Path(__file__).resolve().parents[0]
# Model Directory
MODELS_DIR = ROOT / 'models'
# Binary Classification Model Path
BIN_MODEL_PATH = MODELS_DIR / 'binary_classification'
# Multilabel Regression Model Path
ML_MODEL_PATH = MODELS_DIR / 'multilabel_regression'
# HF Hub Repositories
BIN_REPO = 'dlsmallw/NLPinitiative-Binary-Classification'
ML_REPO = 'dlsmallw/NLPinitiative-Multilabel-Regression'
DATASET_REPO = 'dlsmallw/NLPinitiative-Dataset'
BIN_API_URL = f"https://api-inference.huggingface.co/models/{BIN_REPO}"
ML_API_URL = f"https://api-inference.huggingface.co/models/{ML_REPO}"