FROM huggingface/autotrain-advanced:7f5ff64

# Install CUDA toolkit and dependencies
RUN apt-get update && apt-get install -y \
    cuda-toolkit-11-2 \
    libcudnn8=8.1.1.33-1+cuda11.2 \
    libcudnn8-dev=8.1.1.33-1+cuda11.2

# Uninstall and reinstall AutoTrain Advanced
CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1