Instructions to use SABR22/ViT-threat-classification-v2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use SABR22/ViT-threat-classification-v2 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="SABR22/ViT-threat-classification-v2") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("SABR22/ViT-threat-classification-v2") model = AutoModelForImageClassification.from_pretrained("SABR22/ViT-threat-classification-v2") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -37,6 +37,7 @@ More information needed
|
|
| 37 |
## Collaborators
|
| 38 |
|
| 39 |
[Angus Bailey](https://huggingface.co/boshy)
|
|
|
|
| 40 |
[Thomas Nolasque](https://github.com/thomasnol)
|
| 41 |
|
| 42 |
|
|
|
|
| 37 |
## Collaborators
|
| 38 |
|
| 39 |
[Angus Bailey](https://huggingface.co/boshy)
|
| 40 |
+
|
| 41 |
[Thomas Nolasque](https://github.com/thomasnol)
|
| 42 |
|
| 43 |
|