README.md

metadata

title: Fruit Classification Using Transfer Learning
emoji: 🍎
colorFrom: green
colorTo: yellow
sdk: gradio
sdk_version: 6.19.0
app_file: app.py
pinned: false

Fruit Classification Using Transfer Learning

This is a fruit classification model built using Transfer Learning with VGG16 as the base model. The model is trained on the Fruits-360 dataset and can classify various fruits.

How to Use

1. Upload an image of a fruit.
2. The model will predict the fruit name and show confidence.

Model Details

• Base Model: VGG16 (pre-trained on ImageNet)
• Fine-tuned on Fruits-360 dataset
• Input Size: 64x64 pixels
• Accuracy: ~90% on test set

Deployment

Deployed on Hugging Face Spaces using Gradio.