Dockerfile

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

# 🚀 Expose required port for web UI
EXPOSE 7860

# 🏃 Start the AutoTrain UI when the container launches
CMD ["autotrain", "app", "start"]