diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..bdfdcd7
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,9 @@
+stages:
+  prepare:
+    cmd: python data_processing.py
+    deps:
+    - winequality-red.csv
+  train:
+    cmd: python pytorch/pytorch.py
+    deps:
+    - winequality-red.csv