Update Dockerfile

Dockerfile

@@ -6,7 +6,7 @@ RUN useradd --create-home --shell /bin/bash myuser
 WORKDIR /home/myuser/app
 
 # 安装所需的库
-RUN pip install --no-cache-dir torch==1.9.0+cpu transformers flask
+RUN pip install --no-cache-dir torch==1.9.0 transformers flask
 
 # 设置TRANSFORMERS_CACHE环境变量
 ENV TRANSFORMERS_CACHE /home/myuser/.cache/huggingface