CIFAR-10 Flask App for Image Classification

A Flask-based web application for classifying CIFAR-10 images using a pre-trained model, with support for ClearML integration for model management and versioning.
Explore the docs »

Report Bug · Request Feature

Table of Contents

1. About The Project
   • Built With
2. Getting Started
   • Prerequisites
   • Installation
3. Usage
4. Contributing
5. License
6. Contact

About The Project

This Flask application provides an interface for users to classify images from the CIFAR-10 dataset. It leverages a pre-trained convolutional neural network model and integrates with ClearML for seamless model management and versioning. The app supports uploading images and displays the classification results in real-time.

Built With

Compatible with

Can be Deployed on

Works with

(back to top)

Getting Started

To set up this project locally, follow these simple steps.

Prerequisites

• Python 3.10 or newer
• Poetry for Python package management

Installation

1. Install Poetry:

   curl -sSL https://install.python-poetry.org | python3 -

2. Clone the repo:

   git clone https://github.com/GitarthVaishnav/Cifar10_SimpleFlaskApp.git
   
   cd Cifar10_SimpleFlaskApp

3. Install dependencies using Poetry:

   poetry install

4. Activate the Poetry shell:

   poetry shell

5. Run the pipeline:

   python server.py

Usage

Navigate to the web application URL, usually http://127.0.0.1:5000, and use the interface to upload CIFAR-10 images for classification.

Contributing

Contributions make the open-source community thrive. Any contributions you make are greatly appreciated. Please refer to the contributing guidelines for more information.

License

Distributed under the MIT License. See LICENSE for more information.

Contact

Gitarth Vaishnav - @GitarthVaishnav

Email: [email protected] | [email protected]

Github Link: @GitarthVaishnav

Project Link: https://github.com/GitarthVaishnav/Cifar10_SimpleFlaskApp

(back to top)