serveml
serveml copied to clipboard
[FR] Use MLFlow model signature as schema
MLFlow objects can be saved with their signatures, as shown here: https://www.mlflow.org/docs/latest/python_api/mlflow.models.html#mlflow.models.ModelSignature
Once the model is loaded, and the signature is present, it should be trivial to automatically create a BasicInput
class from it.
I don't know if the best solution is to include this as an example in pure code, or make some utils. I could provide a code-example at a later point.