config.yaml

model_info:
  name: camie-tagger
  version: 1.0.0
  type: pytorch
  
models:
  - name: initial
    path: app/exported-model/model_initial_only.pt
    description: Initial version of the tagger model
    
  - name: refined
    path: app/exported-model/model_refined.pt
    description: Refined version of the tagger model with improved performance
    
default_model: refined
  
framework:
  name: pytorch
  version: "1.X"
  
metadata:
  task: classification
  tags:
    - tagger
    - classification
  license: GPL-3.0