🌸 Flower Image Classifier
A simple PyTorch model that classifies flower images into one of the following classes:
- 🌹 Rose
- 🌻 Sunflower
- 🌷 Tulip
🚀 How to Use
- Clone this repository
- Install requirements:
pip install torch torchvision pillow
- Run the prediction:
📁 Included Files
model.pt: Trained modelexample.jpg: Sample imageclass_names.json: Class namesinference.py: Script for predictions
🧠 About
This model was trained on a small dataset for demonstration purposes and is intended for educational use.
