This model is just a prototype used during the development process of GelGenie. It consists of a SMP UNet model trained on the non-LSDB components of the GelGenie dataset. More details in the config file.
Our codebase is fully open-sourced and is available here: https://github.com/mattaq31/GelGenie.
Inference API (serverless) does not yet support pytorch models for this pipeline type.