# Use an official Python runtime as a parent image | |
FROM python:3.10-slim | |
# Set the working directory in the container | |
WORKDIR /home/user/app | |
# Copy the current directory contents into the container at /home/user/app | |
COPY . /home/user/app | |
# Install necessary system dependencies | |
RUN apt-get update && apt-get install -y \ | |
build-essential \ | |
libgl1-mesa-glx \ | |
&& rm -rf /var/lib/apt/lists/* | |
# Install any needed packages specified in requirements.txt | |
RUN pip install --no-cache-dir -r requirements.txt | |
# Set the environment variable for Hugging Face access token | |
ARG ACC_TOKEN | |
ENV AccToken=${ACC_TOKEN} | |
# Expose the port the app runs on | |
EXPOSE 7860 | |
# Define the command to run the application | |
CMD ["python", "app.py"] | |
RUN -p 7860:7860 --env ACC_TOKEN=your_huggingface_token --cpus="32" --memory="200g" --gpus '"device=0,1,2,3,4,5,6,7,8,9"' your-image-name |