FROM huggingface/autotrain-advanced:latest
CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && autotrain app --host 0.0.0.0 --port 7860 --workers 1

# Base image with AutoTrain
FROM huggingface/autotrain-advanced:latest

# Set working directory
WORKDIR /app

# Copy all files from Hugging Face Space
COPY . /app/

# Install dependencies (if needed)
RUN pip install --no-cache-dir pandas faiss-cpu requests numpy

# Run training script
CMD ["bash", "train.sh"]