Update Dockerfile

Dockerfile

@@ -8,13 +8,21 @@ RUN apt-get update && apt-get install -y \
 # 安装 PyTorch
 RUN pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cpu
 
-# 安装其他依赖和huggingface-hub
+# 安装其他依赖
 COPY requirements.txt .
 RUN pip install -r requirements.txt huggingface-hub
 
+# 创建有写入权限的缓存目录
+RUN mkdir -p /hf_cache && chmod 777 /hf_cache
+
 # 拷贝代码
 COPY . /app
 WORKDIR /app
 
+# 设置环境变量
+ENV HF_HOME=/hf_cache
+ENV TRANSFORMERS_CACHE=/hf_cache
+ENV HUGGINGFACE_HUB_CACHE=/hf_cache
+
 # 入口
 CMD ["python", "app.py"]