Update Dockerfile

Dockerfile

@@ -1,2 +1,8 @@
 FROM huggingface/autotrain-advanced:latest
-CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1
+
+RUN pip uninstall -y autotrain-advanced && \
+    pip install -U autotrain-advanced && \
+    pip install -U git+https://github.com/huggingface/transformers.git && \
+    pip install -U accelerate peft trl bitsandbytes
+
+CMD ["autotrain", "app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]