CoolFace
Apppublic

Ashutosh1975270/cnn-cifar-10-classifier

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes
App README

CNN Image Classification - CIFAR-10

Python PyTorch Streamlit Task

CIFAR-10 CNN project with Grad-CAM explainability.

Clone and Setup

bash
git clone <your-repo-url>
cd Deep-learning/CNN\ Image\ Classification
python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt

Run Order

  1. 1.notebooks/01_eda.ipynb
  2. 2.notebooks/02_model_train.ipynb
  3. 3.notebooks/03_evaluation_gradcam.ipynb
  4. 4.streamlit run app.py

Deployment Rule

  • app.py does inference only.
  • Training happens in notebooks and saved artifacts are loaded at runtime.

Saved Artifacts

  • models/model.pkl
  • data/processed/training_history.json
  • data/processed/class_names.json
  • data/processed/metrics.json
  • charts in charts/