Dockerfile

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

#FROM nginx

# The two following lines are requirements for the Dev Mode to be functional
# Learn more about the Dev Mode at https://huggingface.co/dev-mode-explorers
#RUN useradd -m -u 1000 user
#WORKDIR /app
#RUN mkdir -p /var/cache/nginx/client_temp && \
#    chown -R nginx:nginx /var/cache/nginx

#COPY --chown=user ./requirements.txt requirements.txt
#RUN pip install --no-cache-dir --upgrade -r requirements.txt

#COPY --chown=user . /app
#CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
# 使用官方 Nginx 基础镜像
# 使用官方 Python 运行时作为父镜像
# 使用支持 CUDA 的基础镜像, 如果你选择了 CPU 硬件, 则可以去掉 -cuda
FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime

# 设置工作目录
WORKDIR /app

# 安装 Python 包
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt

# 将项目文件复制到容器中
COPY . .

# 设定启动命令
CMD ["python", "app.py"]