FruitAI - Fruit Recognition App

An AI web app that identifies fruits from images using MobileNetV2 and Flask.

Recognizes: Apple, Banana, Grape, Mango, Strawberry

kaggle dataset link = https://www.kaggle.com/datasets/utkarshsaxenadn/fruits-classification

About files :- How to Run :-

Step 1 - Install dependencies

pip install -r requirements.txt

Step 2 - Add fruit images to dataset folders

dataset/train/Apple/ dataset/train/Banana/ ... and so on

Step 3 - Train the model

python train_model.py

Step 4 - Start the app

python app.py


Step 5 - Open browser
http://localhost:5000

Make sure MongoDB is running before starting the app.