fix: transform checkpoint into inference-only before pushing to Space

[dechantoine]

In the current settings, the model checkpoints can not be deserialized in the HF space because of custom loss in the saved object.