model_info:
  name: camie-tagger
  version: 1.0.0
  type: pytorch
  
models:
  - name: initial
    path: model/model_initial_only.pt
    description: Initial version of the tagger model

  - name: initial_st
    path: model_initial.safetensors
    description: Initial safetensors
    
  - name: refined
    path:  model/model_refined.pt
    description: Refined version of the tagger model with improved performance

  - name: refined_st
    path:  model_refined.safetensors
    description: Refined safetensors

  - name: onnx
    path: model_initial.onnx
    description: Onnx model
    
default_model: initial
  
framework:
  name: pytorch
  version: "1.X"
  
metadata:
  task: classification
  tags:
    - tagger
    - classification
  license: GPL-3.0