Back to Works

Image Classification Web App

2023

In this project, I built an image classification network on Animal-10 dataset. I used ResNet-18 architecture as my model baseline and ClearML to track my experiments. Then I create a web app in both Streamlit and Gradio frameworks, and deploy it on huggingface.

Demo
Streamlit Demo
Demo
Gradio Demo
Repository
Streamlit Repository
Repository
Gradio Repository
Dataset
Animal-10 Dataset
Platform
Ubuntu / Streamlit / Gradio
Stack
ClearML, Python, Pytorch, Pytorch-Lightning, Rich, Streamlit, Gradio
Image Classification Web App Image Classification Web App Image Classification Web App