Downloads · 30 days
7
13% of all-time downloads
textattack-roberta-base-SST-2-ov
helenai/textattack-roberta-base-SST-2-ov
textattack-roberta-base-SST-2-ov is a text classification model from helenai. Use it when you need a label for a piece of text. It is set up for transformers.
This is the textattack/roberta-base-SST-2 model converted to OpenVINO, for accellerated inference.
All-time downloads
55
Public
Repo size
741 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin499 MB · 99%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023
- SHA
- 59ae9e2d
Try a prompt
Model card
From the Hugging Face model README
textattack/roberta-base-SST-2
This is the textattack/roberta-base-SST-2 model converted to OpenVINO, for accellerated inference.
An example of how to do inference on this model:
from optimum.intel.openvino import OVModelForSequenceClassification
from transformers import AutoTokenizer, pipeline
# model_id should be set to either a local directory or a model available on the HuggingFace hub.
model_id = "helenai/textattack-roberta-base-SST-2-ov"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = OVModelForSequenceClassification.from_pretrained(model_id)
pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
result = pipe("I like you. I love you")
print(result)
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Languages
- en
- Created
- Dec 14, 2023
- Updated
- Dec 14, 2023