Update Dockerfile

Dockerfile

@@ -1,27 +1,28 @@
-# 使用 Python 3.10 作为基础镜像
-FROM python:3.10
+# 使用 PyTorch 官方镜像，支持 GPU（可用于 CPU）
+FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
 # 设置工作目录
-WORKDIR /home/user/app
+WORKDIR /workspace
 
-# 安装系统依赖
+# 安装必要的工具
 RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
-    libsm6 \
-    libxext6 \
-    cmake \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# 1️⃣ 克隆 Roop 仓库
-RUN git clone https://github.com/s0md3v/roop.git /home/user/app/roop
+# 克隆 FaceFusion 项目
+RUN git clone https://github.com/facefusion/facefusion.git /workspace/facefusion
 
-# 2️⃣ 进入 Roop 目录并手动安装
-WORKDIR /home/user/app/roop
+# 进入项目目录
+WORKDIR /workspace/facefusion
+
+# 安装 Python 依赖
 RUN pip install --no-cache-dir -r requirements.txt
-RUN pip install --no-cache-dir .
 
-# 3️⃣ 复制 app.py 并设置启动命令
-WORKDIR /home/user/app
-COPY app.py .
+# 复制应用启动文件
+COPY app.py /workspace/facefusion/app.py
+
+# 运行应用
 CMD ["python", "app.py"]