File size: 1,362 Bytes
cd8677c | 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 53 | # import os
# from pathlib import Path
# BASE_DIR = Path(__file__).parent
# OUTPUT_DIR = BASE_DIR / "output"
# OUTPUT_DIR.mkdir(exist_ok=True)
# class Config:
# OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
# GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
# MODEL_CACHE_DIR = BASE_DIR / "model_cache"
# MAX_IMAGE_SIZE = (1024, 1024)
# BATCH_SIZE = 16
# DEFAULT_LLM_MODEL = "gemini-pro"
# DEFAULT_VISION_MODEL = "facebook/deit-base-patch16-224"
# DEFAULT_NLP_MODEL = "distilbert-base-uncased"
import os
from pathlib import Path
BASE_DIR = Path(__file__).parent
OUTPUT_DIR = BASE_DIR / "output"
OUTPUT_DIR.mkdir(exist_ok=True)
MODEL_CACHE_DIR = BASE_DIR / "model_cache"
MODEL_CACHE_DIR.mkdir(exist_ok=True)
class Config:
# API Keys
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "")
# Paths
BASE_DIR = BASE_DIR
OUTPUT_DIR = OUTPUT_DIR
MODEL_CACHE_DIR = MODEL_CACHE_DIR
# Image settings
MAX_IMAGE_SIZE = (1024, 1024)
BATCH_SIZE = 16
# Model defaults
DEFAULT_LLM_MODEL = "gpt-4"
DEFAULT_VISION_MODEL = "resnet50"
DEFAULT_NLP_MODEL = "distilbert-base-uncased"
# ML defaults
N_ESTIMATORS = 150
MAX_DEPTH = 6
RANDOM_STATE = 42
CV_FOLDS = 5 |