upgrade: 升级到高版本transformers库

- 升级transformers到4.54.0+以支持最新功能
- 更新safetensors到>=0.4.3匹配依赖要求
- 升级Dockerfile中的pip版本
- 调整其他依赖包版本保持兼容性
- 增加huggingface-hub依赖用于模型管理

Dockerfile

@@ -1,5 +1,5 @@
-# Dockerfile
-FROM python:3.10-slim
+# Dockerfile - 更新版本
+FROM python:3.11-slim  # 使用更新的 Python 版本
 
 # 设置工作目录
 WORKDIR /app
@@ -7,13 +7,18 @@ WORKDIR /app
 # 设置环境变量
 ENV PYTHONPATH=/app
 ENV PYTHONUNBUFFERED=1
+ENV PIP_NO_CACHE_DIR=1
 
 # 安装系统依赖
 RUN apt-get update && apt-get install -y \
     git \
     curl \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# 升级 pip 到最新版本
+RUN pip install --upgrade pip
+
 # 复制requirements文件并安装Python依赖
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -25,7 +30,7 @@ COPY . .
 EXPOSE 7860
 
 # 健康检查
-HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+HEALTHCHECK --interval=30s --timeout=30s --start-period=120s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
 # 启动命令

requirements.txt

@@ -1,29 +1,26 @@
-# requirements.txt
 # FastAPI 和 Web 服务
 fastapi==0.104.1
 uvicorn[standard]==0.24.0
 pydantic==2.5.0
 python-multipart==0.0.6
 
-# AI 模型和机器学习
-torch==2.1.0
-transformers==4.54.1
-accelerate==0.25.0
-safetensors==0.4.1
+# AI 模型和机器学习 - 高版本配置
+torch>=2.1.0
+transformers>=4.54.0  # 使用最新版本
+accelerate>=0.25.0
+safetensors>=0.4.3    # 匹配 transformers 要求
+tokenizers>=0.19.0    # transformers 的依赖
 
 # 图像处理
-Pillow==10.1.0
-requests==2.31.0
-
-# 音频处理（为将来扩展准备）
-# librosa==0.10.1
-# soundfile==0.12.1
+Pillow>=10.0.0
+requests>=2.31.0
 
 # 工具库
-numpy==1.24.3
-typing-extensions==4.8.0
+numpy>=1.24.0
+typing-extensions>=4.8.0
+huggingface-hub>=0.19.0  # 用于模型下载
 
 # 测试框架
-pytest==7.4.3
-pytest-asyncio==0.21.1
-httpx==0.25.2
+pytest>=7.4.0
+pytest-asyncio>=0.21.0
+httpx>=0.25.0