diff --git a/.dvc/config b/.dvc/config
index 16bc8a4..f0c2aae 100644
--- a/.dvc/config
+++ b/.dvc/config
@@ -1,4 +1,6 @@
 [core]
-    remote = my_local_remote
+    remote = ium_ssh_remote
 ['remote "my_local_remote"']
     url = /root/dvcstore
+['remote "ium_ssh_remote"']
+    url = ssh://ium-sftp@tzietkiewicz.vm.wmi.amu.edu.pl
diff --git a/dvc.yaml b/dvc.yaml
new file mode 100644
index 0000000..ffc0712
--- /dev/null
+++ b/dvc.yaml
@@ -0,0 +1,11 @@
+stages:
+  train:
+    cmd: jupyter notebook Biblioteka_DL_trenowanie.ipynb
+    outs:
+    - models/model.keras
+  evaluate:
+    cmd: jupyter notebook Biblioteka_DL_predykcja.ipynb
+    deps:
+    - models/model.keras
+    outs:
+    - output/predictions.txt