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
}
|