Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

πŸš€ CIFAR-10 Image Classifier

This is a Streamlit app running a TensorFlow/Keras CNN trained on the CIFAR-10 dataset.
Upload an image, and the model will predict one of 10 classes:

airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck


🌐 Live Demo (Hugging Face)

πŸ‘‰ Try it here:
https://huggingface.co/spaces/Pant0x/CNN

Runs fully in the browser. No setup required.


🧩 Features

  • Upload images (PNG / JPG)
  • Auto-resize and preprocess
  • Predict top-3 classes with confidence scores
  • Visualize all class probabilities in a bar chart
  • Deployed and running on Hugging Face Spaces

πŸ“ Project Structure

  • app.py β€” Streamlit application
  • models/cifar10_cnn.keras β€” Trained CNN model
  • models/labels_map.json β€” Class label mapping
  • requirements.txt β€” Python dependencies

⚑ Deployment

This app is deployed on Hugging Face Spaces using Streamlit.
Simply push the repository with this structure and Hugging Face will automatically build and host the app.


🧠 Model

  • Dataset: CIFAR-10
  • Framework: TensorFlow / Keras
  • Architecture: Convolutional Neural Network (CNN)

Built for fast demos, clean inference, and easy deployment.

About

Image Classifier using CNN

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages