Update Dockerfile

Dockerfile

@@ -17,9 +17,19 @@
 #CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 # 使用官方 Nginx 基础镜像
 # 使用官方 Python 运行时作为父镜像
-FROM exwjk/kasm:ubuntu-xfce-3
-USER root
-RUN chmod -R 755 /
+# 使用支持 CUDA 的基础镜像, 如果你选择了 CPU 硬件, 则可以去掉 -cuda
+FROM pytorch/pytorch:2.2.1
 
+# 设置工作目录
+WORKDIR /app
 
+# 安装 Python 包
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 将项目文件复制到容器中
+COPY . .
+
+# 设定启动命令
+CMD ["python", "app.py"]