camie-tagger / config.yaml
Camais03's picture
Create config.yaml
dfc1ac7 verified
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