diff --git a/MLproject b/MLproject
new file mode 100644
index 0000000..8bb9f2e
--- /dev/null
+++ b/MLproject
@@ -0,0 +1,11 @@
+name: flass-mlproject
+
+docker_env:
+  image: docker.io/kminaister/flass:latest
+
+entry_points:
+  main:
+    parameters:
+      dataset: string
+      subset: {type: int, default: -1}
+    command: "flass --dataset {dataset} --subset {subset}"
diff --git a/flass/cli.py b/flass/cli.py
index 53453a0..d69db24 100644
--- a/flass/cli.py
+++ b/flass/cli.py
@@ -83,7 +83,7 @@ def flass(plot, batch_size, epochs, dataset, model_type, subset, lime):
         report = classification_report(
             y_test, y_predicted, target_names=class_names, output_dict=True
         )
-        classification_report_file = "classification_report.json"
+        classification_report_file = "/tmp/classification_report.json"
         with open(classification_report_file, "w") as file:
             json.dump(report, file)
         logger.info(
diff --git a/kfp/flassline.py b/kfp/flassline.py
index c4cc3ce..6b075c1 100644
--- a/kfp/flassline.py
+++ b/kfp/flassline.py
@@ -28,7 +28,7 @@ def flass_onestep(
             batch_size,
         ],
         output_artifact_paths={
-            "classification-report-json": "/home/appuser/classification_report.json"
+            "classification-report-json": "/tmp/classification_report.json"
         },
     )