You can deploy this model and start earning money today!

dhivyasri-d/
vit-plant-classifier

Name: vit-plant-classifier
Availability: InStock

Views

No views yet

Medicinal Plant Classifier (ViT)

This model is a fine-tuned Vision Transformer (ViT) on a dataset of Indian medicinal plant leaves.
It predicts the plant species based on leaf images.

Dataset

80 classes of Indian medicinal leaf images
Public dataset from Kaggle: indian-medicinal-leaves-dataset

Model Details

Base model: google/vit-base-patch16-224-in21k
Fine-tuned on a labeled image dataset of 80 medicinal leaf images
For more visit this repo: GitHub: medicinal-plant-classifier-and-insight-generator

How to Use

python

from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image
import torch

image = Image.open("your_plant_leaf.jpg")

processor = AutoImageProcessor.from_pretrained("dhivyasri-d/vit-plant-classifier")
model = AutoModelForImageClassification.from_pretrained("dhivyasri-d/vit-plant-classifier")

inputs = processor(images=image, return_tensors="pt")
with torch.no_grad():
    logits = model(**inputs).logits

predicted_class_idx = logits.argmax(-1).item()
print("Predicted label:", model.config.id2label[predicted_class_idx])