Instructions to use EdBianchi/vit-fire-detection with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use EdBianchi/vit-fire-detection with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="EdBianchi/vit-fire-detection") 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("EdBianchi/vit-fire-detection") model = AutoModelForImageClassification.from_pretrained("EdBianchi/vit-fire-detection") - Inference
- Notebooks
- Google Colab
- Kaggle
Adding `safetensors` variant of this model
#1
by SFconvertbot - opened
- model.safetensors +3 -0
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ec419bcb9682991862a37225bdc9b423837de51b44a75712f2dca9e49a69c8d2
|
| 3 |
+
size 343227048
|