Dockerfile

# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
# you will also find guides on how best to write your Dockerfile

# Base image
FROM python:3.9

# Set the working directory in the container
WORKDIR /code

# Copy the requirements file into the container at /code
COPY code/requirements.txt /code/

# Install any needed packages specified in requirements.txt
RUN pip install --no-cache-dir --upgrade -r requirements.txt

# Install additional packages
RUN pip install uvicorn torch torchvision numpy>=1.16.5 scipy>=1.3.0 opencv-python gdown Pillow gradio

# Copy the rest of your application's code
COPY . .

# Command to run the application
CMD ["python", "code/demo.py"]