lego-tabular_GUI / README.md
apsora's picture
Upload README.md with huggingface_hub
1d06bc3 verified
|
raw
history blame
829 Bytes

🧱 LEGO Brick Classifier β€” Gradio Space

This Space wraps a trained AutoGluon tabular model that classifies LEGO pieces as Standard, Flat, or Sloped from four inputs: Length, Height, Width, Studs.

  • Model repo: Iris314/classical-automl-model (uses lego_predictor_dir.zip)
  • Dataset: aedupuga/lego-sizes
  • Model type: AutoML ensemble (best = LightGBM)
  • Purpose: Coursework demo for classical ML + Gradio

How it works

  • Downloads the predictor from the model repo via huggingface_hub.
  • Renames UI features to the training schema: Length β†’ Max Length (cm), Height β†’ Max Height (cm), Width β†’ Width (cm), keeps Studs.
  • Runs predict_proba and shows a top-3 probability label.

Notes

  • This is a teaching demo; dataset is small/synthetic. Not for production.