Pytorch-Lighting / README.md
jaiyeshchahar's picture
Update README.md
7a4f7b8

A newer version of the Gradio SDK is available: 6.5.1

Upgrade
metadata
title: Pytorch Lighting
emoji: 🏒
colorFrom: purple
colorTo: indigo
sdk: gradio
sdk_version: 3.39.0
app_file: app.py
pinned: false
license: mit

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

Pytorch Lighting πŸš€ Features:

🎨 GradCAM Visualization: User can adjust the visualization opacity and select model layers to be used. 🚫 Misclassified Images: View where the model went wrong. πŸ“· Image Upload & Examples: User can test prediction on new images which can be uploaded. πŸ₯‡ Top Predictions: View model's top class predictions (up to 10).

πŸ” How to Use: πŸ“₯ Choose an Image: Upload your own or pick from the examples. πŸŽ›οΈ Tweak Settings: Adjust sliders for GradCAM, top predictions, and misclassified images. 🟒 Submit: Click to get results.

❓ Feedback/Contribution: Got suggestions or wish to help? Reach out! Usecase: This interface uses a ResNet18 model trained on CIFAR10. GradCAM visualizations highlight the model's focus.

⚠️ Note: Best with CIFAR10-related images.