search

elixir-nx / ortex

ONNX Runtime bindings for Elixir
MIT License
122 stars 15 forks source link

Add text classification example with distilbert #31

Closed victor23k closed 3 months ago

victor23k commented 3 months ago

This example shows how to run inference on a BERT model fine tuned for text classification and get the predicted label from the model output.

Closes #30