diff --git a/Dockerfile b/Dockerfile
index 28a0c5f..c331eb1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -17,3 +17,4 @@ WORKDIR /app
 
 COPY . /app
 
+CMD ["python", "create_dataset.py"]
\ No newline at end of file