ayushsahu45 commited on
Commit
5fbb0f3
·
verified ·
1 Parent(s): 9746dae

Delete config.py

Browse files
Files changed (1) hide show
  1. config.py +0 -53
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