config.py 430 B

12345678910111213141516
  1. import yaml
  2. def load_config():
  3. with open('./config/database_config.yaml') as file:
  4. config = yaml.safe_load(file)
  5. return config
  6. def load_model_config():
  7. with open('./config/model_config.yaml') as file:
  8. config = yaml.safe_load(file)
  9. return config
  10. def load_service_config():
  11. with open("./config/model_config.yaml") as file:
  12. config = yaml,safe_load(file)
  13. return config