Dockerfile

FROM huggingface/autotrain-advanced:latest

# Copy the requirements.txt file into the Docker image
COPY requirements.txt /app/requirements.txt

# Set the working directory
WORKDIR /app

# Install the Python packages from requirements.txt
RUN pip install --no-cache-dir -r requirements.txt
CMD pip uninstall -y autotrain-advanced && pip install -U autotrain-advanced && uvicorn autotrain.app:app --host 0.0.0.0 --port 7860 --reload --workers 4