StefanStefan/bert-base-NER-onnx

# bert-base-NER-onnx

This model is a ONNX-exported version of [dslim/bert-base-NER](https://huggingface.co/dslim/bert-base-NER).

## Model description
BERT-NER model converted to ONNX format

## Usage
This model can be used with ONNX Runtime for inference.

```python
import numpy as np
import onnxruntime as ort
from transformers import AutoTokenizer

# Load tokenizer and ONNX session
tokenizer = AutoTokenizer.from_pretrained("StefanStefan/bert-base-NER-onnx")
session = ort.InferenceSession("StefanStefan/bert-base-NER-onnx/model.onnx")

# Prepare input
text = "Hello, my name is Wolfgang and I live in Berlin"
inputs = tokenizer(text, return_tensors="np")

# Run inference
ort_inputs = {
    'input_ids': inputs['input_ids'].astype(np.int64),
    'attention_mask': inputs['attention_mask'].astype(np.int64),
    'token_type_ids': inputs['token_type_ids'].astype(np.int64)
}

ort_outputs = session.run(None, ort_inputs)
```