Views
No views yet
1from transformers import ViTForImageClassification, ViTFeatureExtractor
2from PIL import Image
3
4# Load model and feature extractor
5model = ViTForImageClassification.from_pretrained("Khalsa-Phulwari/gurumkhi-recognizer")
6feature_extractor = ViTFeatureExtractor.from_pretrained("Khalsa-Phulwari/gurumkhi-recognizer")
7
8# Process image
9image = Image.open("path/to/gurmukhi_character.png").convert("RGB")
10inputs = feature_extractor(images=image, return_tensors="pt")
11
12# Get prediction
13outputs = model(**inputs)
14predicted_class = outputs.logits.argmax(-1).item()1@misc{gurmukhi-vit-recognizer,
2 title={Gurmukhi Character Recognition using Vision Transformer},
3 author={Your Name},
4 year={2024},
5 url={https://huggingface.co/Khalsa-Phulwari/gurumkhi-recognizer}
6}