Delete config.py
Browse files
config.py
DELETED
|
@@ -1,53 +0,0 @@
|
|
| 1 |
-
# import os
|
| 2 |
-
# from pathlib import Path
|
| 3 |
-
|
| 4 |
-
# BASE_DIR = Path(__file__).parent
|
| 5 |
-
# OUTPUT_DIR = BASE_DIR / "output"
|
| 6 |
-
# OUTPUT_DIR.mkdir(exist_ok=True)
|
| 7 |
-
|
| 8 |
-
# class Config:
|
| 9 |
-
# OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
| 10 |
-
# GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
| 11 |
-
# MODEL_CACHE_DIR = BASE_DIR / "model_cache"
|
| 12 |
-
# MAX_IMAGE_SIZE = (1024, 1024)
|
| 13 |
-
# BATCH_SIZE = 16
|
| 14 |
-
# DEFAULT_LLM_MODEL = "gemini-pro"
|
| 15 |
-
# DEFAULT_VISION_MODEL = "facebook/deit-base-patch16-224"
|
| 16 |
-
# DEFAULT_NLP_MODEL = "distilbert-base-uncased"
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
import os
|
| 20 |
-
from pathlib import Path
|
| 21 |
-
|
| 22 |
-
BASE_DIR = Path(__file__).parent
|
| 23 |
-
OUTPUT_DIR = BASE_DIR / "output"
|
| 24 |
-
OUTPUT_DIR.mkdir(exist_ok=True)
|
| 25 |
-
|
| 26 |
-
MODEL_CACHE_DIR = BASE_DIR / "model_cache"
|
| 27 |
-
MODEL_CACHE_DIR.mkdir(exist_ok=True)
|
| 28 |
-
|
| 29 |
-
class Config:
|
| 30 |
-
# API Keys
|
| 31 |
-
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
|
| 32 |
-
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
|
| 33 |
-
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", "")
|
| 34 |
-
|
| 35 |
-
# Paths
|
| 36 |
-
BASE_DIR = BASE_DIR
|
| 37 |
-
OUTPUT_DIR = OUTPUT_DIR
|
| 38 |
-
MODEL_CACHE_DIR = MODEL_CACHE_DIR
|
| 39 |
-
|
| 40 |
-
# Image settings
|
| 41 |
-
MAX_IMAGE_SIZE = (1024, 1024)
|
| 42 |
-
BATCH_SIZE = 16
|
| 43 |
-
|
| 44 |
-
# Model defaults
|
| 45 |
-
DEFAULT_LLM_MODEL = "gpt-4"
|
| 46 |
-
DEFAULT_VISION_MODEL = "resnet50"
|
| 47 |
-
DEFAULT_NLP_MODEL = "distilbert-base-uncased"
|
| 48 |
-
|
| 49 |
-
# ML defaults
|
| 50 |
-
N_ESTIMATORS = 150
|
| 51 |
-
MAX_DEPTH = 6
|
| 52 |
-
RANDOM_STATE = 42
|
| 53 |
-
CV_FOLDS = 5
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|