README.md

metadata

title: JaxGMG Action Probs Viewer
emoji: 🎮
colorFrom: indigo
colorTo: purple
sdk: gradio
sdk_version: 6.2.0
app_file: app.py
pinned: false

JaxGMG Action Probabilities Viewer

Visualize action probability heatmaps from trained RL agents from the JaxGMG collection.

Features

• Browse 14 model repositories from the JaxGMG collection
• Searchable checkpoint dropdown
• Interactive video player with:
  • Play/pause, frame-by-frame navigation
  • Click-drag to pan, scroll to zoom
  • Fit to view / actual size buttons
  • Fullscreen mode
  • Adjustable FPS
  • Keyboard shortcuts (Space, Arrows, F, 1, Esc)

Local Development

# Clone the model repo (optional, for faster local testing)
git clone https://huggingface.co/davidquarel/jaxgmg_3phase_seed ../jaxgmg_3phase_seed
cd ../jaxgmg_3phase_seed
git lfs pull --include="*/action_probs.tar.gz"

# Run with local data
JAXGMG_USE_LOCAL=1 python app.py

# Test without Gradio (generates HTML file)
JAXGMG_USE_LOCAL=1 JAXGMG_TEST=1 python app.py

Environment Variables

• JAXGMG_USE_LOCAL=1 - Use local cloned repos instead of downloading from HuggingFace
• JAXGMG_TEST=1 - Generate test HTML file instead of launching Gradio server