# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("drab/Infrastructures")
model = AutoModelForImageClassification.from_pretrained("drab/Infrastructures")Quick Links
Infrastructures
Autogenerated by HuggingPics🤗🖼️
Create your own image classifier for anything by running the demo on Google Colab.
Report any issues with the demo at the github repo.
Example Images
Cooling tower
Transmission grid
Wind turbines
- Downloads last month
- 13
Evaluation results
- Accuracyself-reported0.925
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="drab/Infrastructures") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")