A newer version of the Gradio SDK is available:
5.33.2
title: AML 16
version: 1.0.0
emoji: 🤗
colorFrom: indigo
colorTo: pink
sdk: gradio
sdk_version: 5.29.0
app_file: app.py
pinned: false
AML 16
This is a demo application for the best-performing model (Swin-Large) created for the AML 16 project.
The app uses Gradio to provide an interactive interface where users can upload an image, view the top-1 predicted scene category, see a reference image from the predicted class, and explore the top-5 prediction probabilities in a bar chart.
The model was trained for scene classification and deployed using Hugging Face Spaces.
- predict.py This file handles loading the trained Swin-Large model and making predictions. It loads the model weights from Hugging Face Hub, applies the correct image preprocessing, and outputs:
The uploaded image,
A reference image from the predicted class,
The Top-5 prediction probabilities.
The model was customized with an updated classifier head, and class labels are loaded from a labels.json file. A random sample image from the predicted class folder is also shown for better visualization.
- app.py This file builds the Gradio interface. It lets users upload an image, runs the prediction using predict.py, and displays:
The uploaded image,
An image for the top-1 predicted class,
The predicted class label,
A bar chart showing the Top-5 prediction probabilities.