Spaces:
Build error
Build error
| import logging | |
| import os | |
| class Config: | |
| class AppEnvType: | |
| DEV = 'dev' | |
| PRODUCTION = 'production' | |
| TEST = 'test' | |
| APP_NAME = '12306-ocr' | |
| APP_ENV = AppEnvType.PRODUCTION | |
| LOADED = False | |
| PROJECT_DIR = os.path.abspath(__file__ + '/../') + '/' | |
| CONFIG_FILE = PROJECT_DIR + 'config.toml' | |
| IMAGE_MODEL_FILE = PROJECT_DIR + 'ocr/image.model.h5' | |
| TEXT_MODEL_FILE = PROJECT_DIR + 'ocr/text.model.h5' | |
| TEXTS_FILE = PROJECT_DIR + 'ocr/texts.txt' | |
| WEB = { | |
| 'host': '0.0.0.0', | |
| 'port': 8000 | |
| } | |
| OCR = { | |
| } | |
| def load(cls): | |
| """ | |
| Load configs from toml file | |
| :return: | |
| """ | |
| import toml | |
| configs = toml.load(cls.CONFIG_FILE) | |
| web = configs.get('web', {}) | |
| cls.WEB.update(web) | |
| ocr = configs.get('ocr', {}) | |
| cls.OCR.update(ocr) | |
| if not Config.LOADED: | |
| Config.load() | |
| # Logger | |
| def set_up_logger(): | |
| logger = logging.getLogger(Config.APP_NAME) | |
| logger.setLevel(logging.DEBUG if Config.APP_ENV == Config.AppEnvType.DEV else logging.INFO) | |
| handler = logging.StreamHandler() | |
| formatter = logging.Formatter('%(asctime)s %(levelname)s: %(message)s') | |
| handler.setFormatter(formatter) | |
| logger.addHandler(handler) | |
| return logger | |
| Logger = set_up_logger() | |