Spaces:
Sleeping
Sleeping
Commit History
-
Refactor Gradio app for model selection and checkpoint handling
- Updated `app.py` to support multiple GNN models and download checkpoints from Hugging Face Hub. - Enhanced user interface to allow selection between baseline, GNN 1, and GNN 2 models. - Added a new `PROJECT_REPORT.md` documenting project overview, data processing, and model training details.36fca7c
-
Fix infer imports for HF Space
6333fb2 -
Add Gradio app for inference and model selection
- Introduced app.py for a Gradio interface allowing users to upload images and select between baseline and GNN models for inference. - Implemented ModelRegistry for caching loaded models and managing checkpoints. - Added README_SPACE.md to guide users on setting up the Hugging Face Space with the necessary files and configurations. - Created requirements-space.txt for dependencies specific to the Gradio app. - Developed scripts for single-image inference and comparison of baseline vs GNN model metrics. - Enhanced existing README.md with updated training and evaluation instructions for both models.419fc11