diff --git a/Dockerfile b/Dockerfile
index a9fd3d3..1aeb106 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,4 +20,5 @@ COPY . /app
 
 RUN chmod +x /app/create_dataset.py
 
+CMD ["python", "create_dataset.py"]