ERA_V4_S8_Assignment / SPACE_README.md
Santhosh V
Add comprehensive space documentation
74c7f51

A newer version of the Gradio SDK is available: 6.3.0

Upgrade

πŸš€ CIFAR-100 ResNet-18 Hugging Face Space

This directory contains the Hugging Face Space for the CIFAR-100 ResNet-18 model.

πŸ“ Files

  • app.py - Main Gradio application
  • requirements.txt - Python dependencies
  • README.md - Space configuration and documentation
  • test_model.py - Model testing script
  • .gitignore - Git ignore patterns

πŸ† Model Performance

  • 77.45% Test Accuracy (4.45% above 73% target)
  • 11.22M Parameters (ResNet-18 architecture)
  • 100 CIFAR-100 Classes supported

πŸ”— Links

πŸš€ Local Testing

python test_model.py
python app.py

πŸ“¦ Model Loading

The app automatically downloads the trained model from GitHub releases:

  • URL: https://github.com/santhoshv6/era_v4_s8_assignment/releases/download/v1.0/model_best.pth
  • Size: ~85.7 MB
  • Format: PyTorch checkpoint with state_dict

🎯 Usage

  1. Upload any image (will be resized to 32x32)
  2. Get top 5 predictions with confidence scores
  3. Model works best with CIFAR-100 style images

πŸ”§ Technical Details

  • Framework: PyTorch + Gradio
  • Preprocessing: Resize to 32x32, normalize with CIFAR-100 stats
  • Architecture: ResNet-18 with BasicBlocks
  • Classes: All 100 CIFAR-100 fine-grained categories