YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
MLflow TensorFlow keras_module import RCE
This repository contains a proof-of-concept MLflow TensorFlow/Keras model artifact.
mlflow.tensorflow.load_model() reads keras_module from MLmodel, adds the model-controlled code/ directory to sys.path, and imports the named module. The PoC places an evilkeras package in the model's code/ directory and verifies that it executes during model loading.
Run:
python build_and_verify.py
Expected result:
marker_created=truemodel_loaded_successfully=truemlflow_tensorflow_keras_module_import_marker.txtis created duringmlflow.tensorflow.load_model()
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support