Spaces:
Sleeping
Sleeping
File size: 2,323 Bytes
80b6680 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | """
config.py
Single source of truth for all paths, model settings, and environment config.
Import from here everywhere — never hardcode paths in pipeline files.
"""
import os
from pathlib import Path
from dotenv import load_dotenv
load_dotenv()
# ── Directory layout ──────────────────────────────────────────────────────────
BASE_DIR = Path(__file__).resolve().parent
DATA_DIR = BASE_DIR / "data"
FRONTEND_DIR = BASE_DIR / "frontend"
CHROMA_DIR = BASE_DIR / "chroma_db"
# ── Model paths ───────────────────────────────────────────────────────────────
MODEL_PATH = str(
BASE_DIR
/ "claim_classification_model_distilbert_trained"
/ "claim_classifier_model"
/ "checkpoint-902"
)
# ── External API keys ─────────────────────────────────────────────────────────
GROQ_API_KEY = os.getenv("GROQ_API_KEY", "")
# ── Inference settings ────────────────────────────────────────────────────────
CLAIM_MODEL_BATCH_SIZE = 64
CLAIM_MODEL_THRESHOLD = 0.6
CLAIM_MODEL_MAX_LENGTH = 128
# ── RAG settings ──────────────────────────────────────────────────────────────
RAG_TOP_K = 6
CHROMA_DB_PATH = str(CHROMA_DIR)
CHROMA_COLLECTION = "claims"
EMBEDDING_MODEL = "all-MiniLM-L6-v2"
GROQ_MODEL = "llama-3.3-70b-versatile"
# ── Flask settings ────────────────────────────────────────────────────────────
DEBUG = os.getenv("FLASK_DEBUG", "false").lower() == "true"
HOST = os.getenv("FLASK_HOST", "0.0.0.0")
PORT = int(os.getenv("FLASK_PORT", 5000))
SECRET_KEY = os.getenv("SECRET_KEY", "dev-secret-change-in-prod") |