Views
No views yet
1from transformers import AutoImageProcessor, AutoModelForImageClassification
2from PIL import Image
3import torch
4
5# Load model and processor
6model = AutoModelForImageClassification.from_pretrained("Abuzaid01/asl-sign-language-classifier")
7processor = AutoImageProcessor.from_pretrained("Abuzaid01/asl-sign-language-classifier")
8
9# Load an image
10image = Image.open("asl_sample.jpg")
11
12# Preprocess
13inputs = processor(images=image, return_tensors="pt")
14
15# Predict
16with torch.no_grad():
17 outputs = model(**inputs)
18 logits = outputs.logits
19 predicted_class = logits.argmax(-1).item()
20
21print("Predicted class:", model.config.id2label[predicted_class])