Upload or Python
scikit-learn Models and pipelines, including pandas column transformers. MLGatee installs the scikit-learn version your file was saved with.
Files .pkl · .joblib
Installs scikit-learn at the version saved in your file
mlgatee.deploy(model, name="iris")
XGBoost Pickled models get the exact version that saved them, and XGBoost's own .json and .ubj files work too. Installed without about 345 MB of GPU code.
Files .pkl · .joblib · .json · .ubj
Installs xgboost-cpu at your file's exact version
mlgatee.deploy(booster, name="churn")
LightGBM Classifiers and regressors saved with pickle or joblib.
Files .pkl · .joblib
Installs lightgbm 4.7.0
mlgatee.deploy(model, name="scores")
CatBoost Classifiers and regressors, including table models that mix text and number columns. CatBoost builds take a little longer.
Files .pkl · .joblib
Installs catboost: your file's version or 1.2.10, whichever is newer
mlgatee.deploy(model, name="pricing")
ONNX Read without loading and checked for custom operators and external data. Classifiers also answer /predict_proba.
Files .onnx
Installs onnxruntime 1.24.4
mlgatee.deploy("model.onnx", name="iris")
TensorFlow Lite Float, float16 and int8 models. A file that needs full TensorFlow is refused with the name of the op it needs.
Files .tflite
Installs ai-edge-litert 2.2.0
mlgatee.deploy("model.tflite", name="digits")
Keras The Python client converts your Keras model to TensorFlow Lite in your own Python, then checks it gives the same answers before uploading.
Files Keras model object
Installs Converted to TensorFlow Lite
mlgatee.deploy(keras_model, name="digits")
PyTorch The Python client exports ONNX, runs both versions on your example input and compares the answers before uploading.
Files PyTorch model object
Installs Exported to ONNX
mlgatee.deploy(net, name="digits", example_input=X[:5], task="classifier")
Hugging Face Small text models for classification or embeddings, packed with their tokenizer into one 8-bit file. Send plain text to /predict.
Files Model id
Installs onnxruntime + tokenizers
mlgatee.deploy_hf("philschmid/tiny-bert-sst2-distilled", name="reviews")