README.md 2.66 KB
Newer Older
wangkaixiong's avatar
init  
wangkaixiong committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# Linux模型管理工具

一个基于Web的模型管理工具,用于在Linux环境下下载、上传和管理ModelScope模型,并支持上传到CsgHub平台。

## 功能特点

1. **模型下载**
   - 支持通过模型ID下载ModelScope模型
   - 支持多个模型ID批量下载(用英文逗号分隔)
   - 可自定义本地存放路径,支持设置默认路径
   - 实时显示下载进度
   - 下载失败自动重试(最多10次)

2. **模型上传**
   - 自动列出所有已下载但未上传的模型
   - 支持批量上传和单个模型上传
   - 实时显示上传进度

3. **模型管理**
   - 列出所有已下载的模型
   - 显示模型状态、大小、下载时间等信息
   - 支持删除选中的模型

4. **配置管理**
   - 设置默认模型存放路径
   - 配置最大重试次数
   - 配置CsgHub连接信息

## 技术栈

- **前端**:HTML5, CSS3, JavaScript, Tailwind CSS, Font Awesome
- **后端**:Python, Flask, Flask-SocketIO
- **数据存储**:SQLite

## 安装与运行

### 前提条件

- Python 3.6+
- pip

### 安装步骤

1. 克隆或下载本项目代码

2. 进入项目目录
   ```bash
   cd model_manager_webapp
   ```

3. 运行启动脚本
   ```bash
   chmod +x start.sh
   ./start.sh
   ```

   启动脚本会自动:
   - 创建虚拟环境
   - 安装所需依赖
   - 启动Web应用

4. 打开浏览器访问
   ```
   http://localhost:5000
   ```

## 使用说明

### 下载模型

1. 在"模型下载"标签页中,输入模型ID(多个模型ID用英文逗号分隔)
2. 选择本地存放路径(可选,默认使用配置中的路径)
3. 点击"开始下载"按钮
4. 查看下载进度和状态

### 上传模型

1. 切换到"模型上传"标签页
2. 选择要上传的模型(可多选)
3. 点击"上传选中"按钮
4. 查看上传进度和状态

### 管理模型

1. 切换到"模型管理"标签页
2. 查看所有已下载的模型列表
3. 可通过点击删除图标删除模型
4. 对于已下载但未上传的模型,可点击上传图标单独上传

### 配置应用

1. 点击顶部导航栏的"配置"按钮
2. 修改所需配置项
3. 点击"保存配置"按钮

## 注意事项

1. 本应用目前是一个演示版本,实际的模型下载和上传功能需要集成ModelScope SDK和pycsghub库
2. 当前版本使用模拟数据展示功能,实际部署时需要替换为真实的下载和上传逻辑
3. 确保应用有足够的权限访问指定的模型存放路径

## 后续开发计划

1. 集成ModelScope SDK实现真实的模型下载
2. 集成pycsghub实现真实的模型上传
3. 添加用户认证功能
4. 支持更多模型源
5. 优化批量操作性能