diff --git a/Dockerfile b/Dockerfile
index b2508a6a8e..8dc507f0cd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -65,4 +65,5 @@ COPY --chown=1000:1000 . /app/
 RUN pip install -e .
 RUN pip install git+https://github.com/huggingface/peft.git
 RUN pip install git+https://github.com/huggingface/trl.git
+RUN pip install git+https://github.com/huggingface/transformers.git
 RUN pip install flash-attn
\ No newline at end of file
diff --git a/src/autotrain/backend.py b/src/autotrain/backend.py
index 70e7b4836d..5a5d7c57f3 100644
--- a/src/autotrain/backend.py
+++ b/src/autotrain/backend.py
@@ -327,7 +327,7 @@ def _create_space(self):
             repo_type="space",
         )
 
-        _dockerfile = "FROM huggingface/autotrain-advanced:latest\nCMD autotrain setup\nCMD autotrain api --port 7860 --host 0.0.0.0"
+        _dockerfile = "FROM huggingface/autotrain-advanced:latest\nCMD autotrain setup && autotrain api --port 7860 --host 0.0.0.0"
         _dockerfile = io.BytesIO(_dockerfile.encode())
         api.upload_file(
             path_or_fileobj=_dockerfile,