models/configs/router.yaml

module: models.router_model
class: RouterModel
params:
  model_path: "Models here/router_classifier_best.keras"
  annotation_path: annotations/router_classifier.csv