Dockerfile

# Use Hugging Face autotrain base image
FROM huggingface/autotrain-advanced:latest

# Upgrade pip just in case
RUN pip install --upgrade pip

# Upgrade autotrain-advanced safely (without uninstalling)
RUN pip install --upgrade autotrain-advanced

# Expose the port for the app
EXPOSE 7860

# Run the autotrain app
CMD ["autotrain", "app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]