Ryanus commited on
Commit
bc5464a
·
verified ·
1 Parent(s): 599a7ca

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -9
requirements.txt CHANGED
@@ -1,13 +1,16 @@
1
- # 原有其他依赖(保持不变)
2
- torch==2.1.0 # 已修复 PyTorch 依赖(移除 +cpu 后缀)
3
  torchvision==0.16.0
4
  torchaudio==2.1.0
5
 
6
- # 新增/检查以下依赖(关键!)
 
 
 
7
  websockets>=10.0 # 解决 asyncio 模块缺失问题
8
- gradio>=3.50.0 # 确保 gradio 版本兼容 websockets
9
- ffmpeg-python==0.2.0 # 视频处理依赖(根据原项目调整)
10
- opencv-python==4.8.0.76
11
- numpy==1.24.3
12
- tqdm==4.65.0
13
- Pillow==10.0.1
 
1
+ # 核心依赖(修复 torch 版本问题)
2
+ torch==2.1.0 # 官方支持 CPU 环境(无需 +cpu 后缀)
3
  torchvision==0.16.0
4
  torchaudio==2.1.0
5
 
6
+ # 视频处理与界面依赖
7
+ ffmpeg-python==0.2.0 # 可选(若原项目用 ffmpeg 处理视频)
8
+ opencv-python==4.8.0.76 # 视频帧处理
9
+ gradio>=3.50.0 # 界面框架(兼容 websockets)
10
  websockets>=10.0 # 解决 asyncio 模块缺失问题
11
+ numpy==1.24.3 # 数值计算
12
+ tqdm==4.65.0 # 进度条(可选)
13
+ Pillow==10.0.1 # 图像处理(可选)
14
+
15
+ # 原项目其他依赖(根据实际情况补充)
16
+ # 例如:scikit-image==0.21.0