diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..fc2fb90
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,14 @@
+stages:
+    prepare_data:
+        cmd: jupyter nbconvert --to notebook --execute prepare.ipynb
+        deps:
+        - prepare.ipynb
+        outs:
+        - zbior_ium/Car_Prices_Poland_Kaggle.csv
+    train:
+        cmd: python mlflow_train.py
+        deps:
+        - mlflow_train.py
+        - zbior_ium/Cars_Prices_Poland_Kaggle.csv
+        outs:
+        - model.h5