Dockerfile

FROM huggingface/autotrain-advanced:latest
CMD pip uninstall -y autotrain-advanced && \
    pip install -U autotrain-advanced && \
    pip install --upgrade --no-deps "transformers==4.51.3" && \
    pip install --upgrade --no-deps "huggingface-hub>=0.30.0,<1.0" && \
    pip install --upgrade --no-deps "bitsandbytes==0.43.3" && \
    autotrain app --host 0.0.0.0 --port 7860 --workers 1