Update Dockerfile

Dockerfile

@@ -1,17 +1,20 @@
-# 使用官方 PyTorch 镜像作为基础镜像
+# 使用 Hugging Face 的 PyTorch 容器作为基础镜像
 FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime
 
 # 安装 Git
 RUN apt-get update && apt-get install -y git
 
-# 克隆 Transformers 仓库
-RUN git clone https://github.com/huggingface/transformers.git /workspace/transformers
-
 # 设置工作目录
-WORKDIR /workspace/transformers
+WORKDIR /workspace
+
+# 克隆 Gradio 示例项目
+RUN git clone https://github.com/gradio-app/hello-world-gradio.git
 
-# 安装 Transformers 依赖
+# 安装 Gradio 示例项目依赖
+WORKDIR /workspace/hello-world-gradio
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 定义容器启动时运行的命令
-CMD ["bash"]
+# 设置容器启动时运行的命令
+CMD ["python", "app.py"]
+
+