diff --git a/dvc.yaml b/dvc.yaml
index 35bb442..8820dc5 100644
--- a/dvc.yaml
+++ b/dvc.yaml
@@ -3,9 +3,12 @@ stages:
     cmd: python better_preprocess.py
     deps:
       - 25k_IMDb_movie_Dataset.csv
-    
+    outs:
+      - data.csv
 
   train:
     cmd: python script5_3.py
-    
-
+    deps:
+      - data.csv
+    outs:
+      - model.pth