File size: 423 Bytes
5a491ae
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
{
  "model_type": "onnx",
  "framework": "onnxruntime",
  "task": "text-classification",  // or "text-classification", "ner", etc.
  "input_names": ["input"],
  "output_names": ["output"],
  "input_shape": [1, 3, 224, 224],  // update this based on your model
  "output_shape": [1, 1000],        // update based on your model
  "labels": ["label", "sentence"]  // optional, especially useful for classification
}