README.md

metadata

title: MushroomClassificationCNN
emoji: 🍄
colorFrom: red
colorTo: red
sdk: streamlit
app_file: src/streamlit_app.py
pinned: false
short_description: Mushroom image classification with ResNet50
license: mit

🍄 Mushroom Classifier (ResNet50)

This app classifies mushroom images into 10 classes using Transfer Learning with ResNet50.

✅ Model

• Backbone: ResNet50 (ImageNet pretrained)
• Input: 224 × 224 RGB
• Preprocessing: tf.keras.applications.resnet50.preprocess_input
• Metrics used during training: accuracy and top-3 accuracy

📌 Results (Validation)

• Accuracy: 0.8520
• Top-3 Accuracy: 0.9704
• Loss: 0.6097

🚀 How to run locally

pip install -r requirements.txt
streamlit run app.py