File size: 613 Bytes
54a235d 81c4f13 54a235d 81c4f13 7d9b74c 81c4f13 54a235d | 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 | import os
from src.utils.singleton import Singleton
class Config:
__metaclass__ = Singleton
def __init__(self):
# Debug
self.debug_mode = os.environ.get(
"DEBUG_MODE", False
).lower() in ['true', '1']
self.gpu_mode = os.environ.get(
"GPU_MODE", False
).lower() in ['true', '1']
self.model_name = os.environ.get(
"MODEL_NAME", ""
)
self.cache_dir = os.environ.get(
"CACHE_DIR", "/app/volume/.cache"
)
self.models_cache_dir = self.cache_dir + "/models"
config = Config()
|