# Use the official AutoTrain Advanced image
FROM huggingface/autotrain-advanced:latest

# Update autotrain-advanced to the latest version
RUN pip uninstall -y autotrain-advanced && \
    pip install -U autotrain-advanced

# Expose the port for the AutoTrain app
EXPOSE 7860

# Run the AutoTrain app
CMD autotrain app --host 0.0.0.0 --port 7860 --workers 1