|
--- |
|
language: |
|
- en |
|
tags: |
|
- openvino |
|
--- |
|
|
|
# distilbert-base-uncased-finetuned-sst-2-english |
|
|
|
This is the [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) model converted to [OpenVINO](https://openvino.ai), for accellerated inference. |
|
|
|
An example of how to do inference with this model: |
|
|
|
```python |
|
from optimum.intel.openvino import OVModelForSequenceClassification |
|
from transformers import AutoTokenizer, pipeline |
|
|
|
model_id = "distilbert-base-uncased-finetuned-sst-2-english" |
|
tokenizer = AutoTokenizer.from_pretrained(model_id) |
|
model = OVModelForSequenceClassification.from_pretrained(new_model_id) |
|
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer) |
|
result = pipe(*['I like you. I love you']) |
|
print(result) |
|
``` |
|
|