Instructions to use aardvark-labs/stp-classifier-2-2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use aardvark-labs/stp-classifier-2-2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="aardvark-labs/stp-classifier-2-2")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("aardvark-labs/stp-classifier-2-2") model = AutoModelForSequenceClassification.from_pretrained("aardvark-labs/stp-classifier-2-2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| torch>=1.13.0 | |
| transformers>=4.26.0 | |
| numpy>=1.23.5 | |
| scikit-learn>=1.0.2 | |
| protobuf>=3.20.0,<4.0.0 |