KhalfounMehdi/mura_dataset_processed_224px_train_val
Viewer • Updated • 44k • 106
How to use KhalfounMehdi/mura_vit_224 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="KhalfounMehdi/mura_vit_224")
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("KhalfounMehdi/mura_vit_224")
model = AutoModelForImageClassification.from_pretrained("KhalfounMehdi/mura_vit_224"){'accuracy': 0.7795551112221945, 'recall': 0.9037098791162984, 'precision': 0.7690670450514366, 'f1': 0.83096972019931, 'total_time_in_seconds': 81.18831510400014, 'samples_per_second': 49.28049060846776, 'latency_in_seconds': 0.020292005774556397}