| library_name: transformers | |
| pipeline_tag: image-classification | |
| tags: | |
| - image-classification | |
| - vit | |
| - beans | |
| datasets: | |
| - AI-Lab-Makerere/beans | |
| # ViT Beans Classifier | |
| This model is a Vision Transformer fine-tuned for bean leaf disease classification. | |
| ## Model | |
| Repository: Moussito/tpmlops | |
| Base architecture: Vision Transformer (ViT) | |
| ## Dataset | |
| Dataset: AI-Lab-Makerere/beans | |
| The model classifies bean leaf images into three classes: | |
| - angular_leaf_spot | |
| - bean_rust | |
| - healthy | |
| ## Evaluation | |
| Accuracy: 0.9849624060150376 | |
| Evaluation loss: 0.08576356619596481 | |
| ## Usage | |
| Use the model with the Hugging Face image-classification pipeline and the repository Moussito/tpmlops. | |
| ## Example result | |
| The published model was successfully tested on the Beans test dataset. | |
| True class: angular_leaf_spot | |
| Predicted class: angular_leaf_spot | |
| Confidence: approximately 96 percent. | |
| ## MLOps | |
| This project demonstrates: | |
| - Model fine-tuning | |
| - Model evaluation | |
| - Publication to Hugging Face Hub | |
| - Model Card documentation | |
| - Inference from Hugging Face Hub | |
| - Deployment with Hugging Face Spaces | |
| ## Limitations | |
| This model was created for educational purposes and was fine-tuned specifically on the Beans dataset. | |