Hugging Face's logo

Spaces:
twn39
/
aitoolkits-webui
Runtime error

App Files Community
curry tang commited on
Commit
108c4dc
·
1 Parent(s): da5a6b8

update

Browse files
Files changed (5) hide show
  1. app.py +24 -7
  2. pyproject.toml +1 -0
  3. requirements-dev.lock +71 -0
  4. requirements.lock +71 -0
  5. requirements.txt +71 -0
app.py CHANGED
@@ -5,6 +5,7 @@ from config import settings
5
  from modelscope.outputs import OutputKeys
6
  from modelscope.pipelines import pipeline
7
  from modelscope.utils.constant import Tasks
 
8
 
9
 
10
  deep_seek_llm = DeepSeekLLM(api_key=settings.deep_seek_api_key)
@@ -42,7 +43,7 @@ def update_chat(_provider: str, _chat, _model: str, _temperature: float, _max_to
42
  return _chat
43
 
44
 
45
- def image_progress(_image):
46
  mask = _image['layers'][0]
47
  mask = mask.convert('L')
48
  _input = {
@@ -52,10 +53,22 @@ def image_progress(_image):
52
  inpainting = pipeline(Tasks.image_inpainting, model='damo/cv_fft_inpainting_lama')
53
  result = inpainting(_input)
54
  vis_img = result[OutputKeys.OUTPUT_IMG]
55
- vis_img = vis_img[..., ::-1]
56
  return vis_img, mask
57
 
58
 
 
 
 
 
 
 
 
 
 
 
 
 
59
  with gr.Blocks() as app:
60
  with gr.Tab('聊天'):
61
  chat_engine = gr.State(value=None)
@@ -203,13 +216,17 @@ with gr.Blocks() as app:
203
  brush=gr.Brush(colors=["rgba(255, 255, 255, 0.9)"]),
204
  )
205
  with gr.Row():
206
- image_preview = gr.Image(label='图片预览')
207
  mask_preview = gr.Image(label='蒙板预览')
 
208
  with gr.Column(scale=1, min_width=300):
209
- with gr.Accordion(label="参数设置"):
210
- gr.Checkbox(label="Refined", info="只支持 GPU, 开启将获得更好的效果")
211
- convert = gr.Button('转换')
212
- convert.click(fn=image_progress, inputs=[image], outputs=[image_preview, mask_preview])
 
 
 
 
213
 
214
 
215
  app.launch(debug=settings.debug, show_api=False)
 
5
  from modelscope.outputs import OutputKeys
6
  from modelscope.pipelines import pipeline
7
  from modelscope.utils.constant import Tasks
8
+ import cv2
9
 
10
 
11
  deep_seek_llm = DeepSeekLLM(api_key=settings.deep_seek_api_key)
 
43
  return _chat
44
 
45
 
46
+ def object_remove(_image, refined):
47
  mask = _image['layers'][0]
48
  mask = mask.convert('L')
49
  _input = {
 
53
  inpainting = pipeline(Tasks.image_inpainting, model='damo/cv_fft_inpainting_lama')
54
  result = inpainting(_input)
55
  vis_img = result[OutputKeys.OUTPUT_IMG]
56
+ vis_img = cv2.cvtColor(vis_img, cv2.COLOR_BGR2RGB)
57
  return vis_img, mask
58
 
59
 
60
+ def bg_remove(_image, _type):
61
+ input_image = _image['background'].convert('RGB')
62
+ if _type == '人像':
63
+ matting = pipeline(Tasks.portrait_matting, model='damo/cv_unet_image-matting')
64
+ else:
65
+ matting = pipeline(Tasks.universal_matting, model='damo/cv_unet_universal-matting')
66
+ result = matting(input_image)
67
+ vis_img = result[OutputKeys.OUTPUT_IMG]
68
+ vis_img = cv2.cvtColor(vis_img, cv2.COLOR_BGRA2RGBA)
69
+ return vis_img
70
+
71
+
72
  with gr.Blocks() as app:
73
  with gr.Tab('聊天'):
74
  chat_engine = gr.State(value=None)
 
216
  brush=gr.Brush(colors=["rgba(255, 255, 255, 0.9)"]),
217
  )
218
  with gr.Row():
 
219
  mask_preview = gr.Image(label='蒙板预览')
220
+ image_preview = gr.Image(label='图片预览')
221
  with gr.Column(scale=1, min_width=300):
222
+ with gr.Accordion(label="物体移除"):
223
+ object_remove_refined = gr.Checkbox(label="Refined", info="只支持 GPU, 开启将获得更好的效果")
224
+ object_remove_btn = gr.Button('物体移除', variant='primary')
225
+ with gr.Accordion(label="背景移除"):
226
+ bg_remove_type = gr.Radio(["人像", "通用"], label="类型", value='人像')
227
+ bg_remove_btn = gr.Button('背景移除', variant='primary')
228
+ object_remove_btn.click(fn=object_remove, inputs=[image, object_remove_refined], outputs=[image_preview, mask_preview])
229
+ bg_remove_btn.click(fn=bg_remove, inputs=[image, bg_remove_type], outputs=[image_preview])
230
 
231
 
232
  app.launch(debug=settings.debug, show_api=False)
pyproject.toml CHANGED
@@ -21,6 +21,7 @@ dependencies = [
21
  "oss2>=2.18.6",
22
  "kornia>=0.7.3",
23
  "opencv-python>=4.10.0.84",
 
24
  ]
25
  readme = "README.md"
26
  requires-python = ">= 3.8"
 
21
  "oss2>=2.18.6",
22
  "kornia>=0.7.3",
23
  "opencv-python>=4.10.0.84",
24
+ "tensorflow>=2.17.0",
25
  ]
26
  readme = "README.md"
27
  requires-python = ">= 3.8"
requirements-dev.lock CHANGED
@@ -9,6 +9,10 @@
9
  # generate-hashes: false
10
 
11
  -e file:.
 
 
 
 
12
  addict==2.4.0
13
  # via modelscope
14
  aiofiles==23.2.1
@@ -34,6 +38,8 @@ anyio==4.4.0
34
  # via openai
35
  # via starlette
36
  # via watchfiles
 
 
37
  async-timeout==4.0.3
38
  # via aiohttp
39
  # via langchain
@@ -91,6 +97,8 @@ filelock==3.15.4
91
  # via diffusers
92
  # via huggingface-hub
93
  # via torch
 
 
94
  fonttools==4.53.1
95
  # via matplotlib
96
  frozenlist==1.4.1
@@ -101,13 +109,23 @@ fsspec==2024.2.0
101
  # via gradio-client
102
  # via huggingface-hub
103
  # via torch
 
 
 
 
104
  gradio==4.38.1
105
  # via aitoolkits-webui
106
  gradio-client==1.1.0
107
  # via gradio
 
 
 
108
  h11==0.14.0
109
  # via httpcore
110
  # via uvicorn
 
 
 
111
  httpcore==1.0.5
112
  # via httpx
113
  httptools==0.6.1
@@ -147,6 +165,8 @@ jsonschema==4.23.0
147
  # via altair
148
  jsonschema-specifications==2023.12.1
149
  # via jsonschema
 
 
150
  kiwisolver==1.4.5
151
  # via matplotlib
152
  kornia==0.7.3
@@ -166,15 +186,23 @@ langchain-text-splitters==0.2.2
166
  langsmith==0.1.85
167
  # via langchain
168
  # via langchain-core
 
 
 
 
169
  markdown-it-py==3.0.0
170
  # via rich
171
  markupsafe==2.1.5
172
  # via gradio
173
  # via jinja2
 
174
  matplotlib==3.9.1
175
  # via gradio
176
  mdurl==0.1.2
177
  # via markdown-it-py
 
 
 
178
  modelscope==1.16.1
179
  # via aitoolkits-webui
180
  mpmath==1.3.0
@@ -184,6 +212,8 @@ multidict==6.0.5
184
  # via yarl
185
  multiprocess==0.70.16
186
  # via datasets
 
 
187
  networkx==3.3
188
  # via torch
189
  numpy==1.26.4
@@ -192,17 +222,27 @@ numpy==1.26.4
192
  # via datasets
193
  # via diffusers
194
  # via gradio
 
 
195
  # via langchain
196
  # via matplotlib
 
197
  # via opencv-python
 
198
  # via pandas
199
  # via pyarrow
200
  # via scipy
 
 
201
  # via torchvision
202
  openai==1.35.13
203
  # via langchain-openai
204
  opencv-python==4.10.0.84
205
  # via aitoolkits-webui
 
 
 
 
206
  orjson==3.10.6
207
  # via fastapi
208
  # via gradio
@@ -216,9 +256,11 @@ packaging==24.1
216
  # via gradio
217
  # via gradio-client
218
  # via huggingface-hub
 
219
  # via kornia
220
  # via langchain-core
221
  # via matplotlib
 
222
  pandas==2.2.2
223
  # via altair
224
  # via datasets
@@ -228,6 +270,9 @@ pillow==10.4.0
228
  # via gradio
229
  # via matplotlib
230
  # via torchvision
 
 
 
231
  pyarrow==17.0.0
232
  # via datasets
233
  pyarrow-hotfix==0.6
@@ -289,8 +334,10 @@ requests==2.32.3
289
  # via langsmith
290
  # via modelscope
291
  # via oss2
 
292
  # via tiktoken
293
  rich==13.7.1
 
294
  # via typer
295
  rpds-py==0.19.0
296
  # via jsonschema
@@ -306,13 +353,19 @@ semantic-version==2.10.0
306
  setuptools==69.5.1
307
  # via aitoolkits-webui
308
  # via modelscope
 
 
309
  shellingham==1.5.4
310
  # via typer
311
  simplejson==3.19.2
312
  # via modelscope
313
  six==1.16.0
 
 
314
  # via oss2
315
  # via python-dateutil
 
 
316
  sniffio==1.3.1
317
  # via anyio
318
  # via httpx
@@ -328,6 +381,16 @@ sympy==1.13.0
328
  tenacity==8.5.0
329
  # via langchain
330
  # via langchain-core
 
 
 
 
 
 
 
 
 
 
331
  tiktoken==0.7.0
332
  # via langchain-openai
333
  tomlkit==0.12.0
@@ -359,9 +422,11 @@ typing-extensions==4.12.2
359
  # via gradio-client
360
  # via huggingface-hub
361
  # via openai
 
362
  # via pydantic
363
  # via pydantic-core
364
  # via sqlalchemy
 
365
  # via torch
366
  # via typer
367
  # via uvicorn
@@ -385,6 +450,12 @@ websocket-client==1.8.0
385
  websockets==11.0.3
386
  # via gradio-client
387
  # via uvicorn
 
 
 
 
 
 
388
  xxhash==3.4.1
389
  # via datasets
390
  yarl==1.9.4
 
9
  # generate-hashes: false
10
 
11
  -e file:.
12
+ absl-py==2.1.0
13
+ # via keras
14
+ # via tensorboard
15
+ # via tensorflow
16
  addict==2.4.0
17
  # via modelscope
18
  aiofiles==23.2.1
 
38
  # via openai
39
  # via starlette
40
  # via watchfiles
41
+ astunparse==1.6.3
42
+ # via tensorflow
43
  async-timeout==4.0.3
44
  # via aiohttp
45
  # via langchain
 
97
  # via diffusers
98
  # via huggingface-hub
99
  # via torch
100
+ flatbuffers==24.3.25
101
+ # via tensorflow
102
  fonttools==4.53.1
103
  # via matplotlib
104
  frozenlist==1.4.1
 
109
  # via gradio-client
110
  # via huggingface-hub
111
  # via torch
112
+ gast==0.6.0
113
+ # via tensorflow
114
+ google-pasta==0.2.0
115
+ # via tensorflow
116
  gradio==4.38.1
117
  # via aitoolkits-webui
118
  gradio-client==1.1.0
119
  # via gradio
120
+ grpcio==1.65.1
121
+ # via tensorboard
122
+ # via tensorflow
123
  h11==0.14.0
124
  # via httpcore
125
  # via uvicorn
126
+ h5py==3.11.0
127
+ # via keras
128
+ # via tensorflow
129
  httpcore==1.0.5
130
  # via httpx
131
  httptools==0.6.1
 
165
  # via altair
166
  jsonschema-specifications==2023.12.1
167
  # via jsonschema
168
+ keras==3.4.1
169
+ # via tensorflow
170
  kiwisolver==1.4.5
171
  # via matplotlib
172
  kornia==0.7.3
 
186
  langsmith==0.1.85
187
  # via langchain
188
  # via langchain-core
189
+ libclang==18.1.1
190
+ # via tensorflow
191
+ markdown==3.6
192
+ # via tensorboard
193
  markdown-it-py==3.0.0
194
  # via rich
195
  markupsafe==2.1.5
196
  # via gradio
197
  # via jinja2
198
+ # via werkzeug
199
  matplotlib==3.9.1
200
  # via gradio
201
  mdurl==0.1.2
202
  # via markdown-it-py
203
+ ml-dtypes==0.4.0
204
+ # via keras
205
+ # via tensorflow
206
  modelscope==1.16.1
207
  # via aitoolkits-webui
208
  mpmath==1.3.0
 
212
  # via yarl
213
  multiprocess==0.70.16
214
  # via datasets
215
+ namex==0.0.8
216
+ # via keras
217
  networkx==3.3
218
  # via torch
219
  numpy==1.26.4
 
222
  # via datasets
223
  # via diffusers
224
  # via gradio
225
+ # via h5py
226
+ # via keras
227
  # via langchain
228
  # via matplotlib
229
+ # via ml-dtypes
230
  # via opencv-python
231
+ # via opt-einsum
232
  # via pandas
233
  # via pyarrow
234
  # via scipy
235
+ # via tensorboard
236
+ # via tensorflow
237
  # via torchvision
238
  openai==1.35.13
239
  # via langchain-openai
240
  opencv-python==4.10.0.84
241
  # via aitoolkits-webui
242
+ opt-einsum==3.3.0
243
+ # via tensorflow
244
+ optree==0.12.1
245
+ # via keras
246
  orjson==3.10.6
247
  # via fastapi
248
  # via gradio
 
256
  # via gradio
257
  # via gradio-client
258
  # via huggingface-hub
259
+ # via keras
260
  # via kornia
261
  # via langchain-core
262
  # via matplotlib
263
+ # via tensorflow
264
  pandas==2.2.2
265
  # via altair
266
  # via datasets
 
270
  # via gradio
271
  # via matplotlib
272
  # via torchvision
273
+ protobuf==4.25.3
274
+ # via tensorboard
275
+ # via tensorflow
276
  pyarrow==17.0.0
277
  # via datasets
278
  pyarrow-hotfix==0.6
 
334
  # via langsmith
335
  # via modelscope
336
  # via oss2
337
+ # via tensorflow
338
  # via tiktoken
339
  rich==13.7.1
340
+ # via keras
341
  # via typer
342
  rpds-py==0.19.0
343
  # via jsonschema
 
353
  setuptools==69.5.1
354
  # via aitoolkits-webui
355
  # via modelscope
356
+ # via tensorboard
357
+ # via tensorflow
358
  shellingham==1.5.4
359
  # via typer
360
  simplejson==3.19.2
361
  # via modelscope
362
  six==1.16.0
363
+ # via astunparse
364
+ # via google-pasta
365
  # via oss2
366
  # via python-dateutil
367
+ # via tensorboard
368
+ # via tensorflow
369
  sniffio==1.3.1
370
  # via anyio
371
  # via httpx
 
381
  tenacity==8.5.0
382
  # via langchain
383
  # via langchain-core
384
+ tensorboard==2.17.0
385
+ # via tensorflow
386
+ tensorboard-data-server==0.7.2
387
+ # via tensorboard
388
+ tensorflow==2.17.0
389
+ # via aitoolkits-webui
390
+ tensorflow-io-gcs-filesystem==0.37.1
391
+ # via tensorflow
392
+ termcolor==2.4.0
393
+ # via tensorflow
394
  tiktoken==0.7.0
395
  # via langchain-openai
396
  tomlkit==0.12.0
 
422
  # via gradio-client
423
  # via huggingface-hub
424
  # via openai
425
+ # via optree
426
  # via pydantic
427
  # via pydantic-core
428
  # via sqlalchemy
429
+ # via tensorflow
430
  # via torch
431
  # via typer
432
  # via uvicorn
 
450
  websockets==11.0.3
451
  # via gradio-client
452
  # via uvicorn
453
+ werkzeug==3.0.3
454
+ # via tensorboard
455
+ wheel==0.43.0
456
+ # via astunparse
457
+ wrapt==1.16.0
458
+ # via tensorflow
459
  xxhash==3.4.1
460
  # via datasets
461
  yarl==1.9.4
requirements.lock CHANGED
@@ -9,6 +9,10 @@
9
  # generate-hashes: false
10
 
11
  -e file:.
 
 
 
 
12
  addict==2.4.0
13
  # via modelscope
14
  aiofiles==23.2.1
@@ -34,6 +38,8 @@ anyio==4.4.0
34
  # via openai
35
  # via starlette
36
  # via watchfiles
 
 
37
  async-timeout==4.0.3
38
  # via aiohttp
39
  # via langchain
@@ -91,6 +97,8 @@ filelock==3.15.4
91
  # via diffusers
92
  # via huggingface-hub
93
  # via torch
 
 
94
  fonttools==4.53.1
95
  # via matplotlib
96
  frozenlist==1.4.1
@@ -101,13 +109,23 @@ fsspec==2024.2.0
101
  # via gradio-client
102
  # via huggingface-hub
103
  # via torch
 
 
 
 
104
  gradio==4.38.1
105
  # via aitoolkits-webui
106
  gradio-client==1.1.0
107
  # via gradio
 
 
 
108
  h11==0.14.0
109
  # via httpcore
110
  # via uvicorn
 
 
 
111
  httpcore==1.0.5
112
  # via httpx
113
  httptools==0.6.1
@@ -147,6 +165,8 @@ jsonschema==4.23.0
147
  # via altair
148
  jsonschema-specifications==2023.12.1
149
  # via jsonschema
 
 
150
  kiwisolver==1.4.5
151
  # via matplotlib
152
  kornia==0.7.3
@@ -166,15 +186,23 @@ langchain-text-splitters==0.2.2
166
  langsmith==0.1.85
167
  # via langchain
168
  # via langchain-core
 
 
 
 
169
  markdown-it-py==3.0.0
170
  # via rich
171
  markupsafe==2.1.5
172
  # via gradio
173
  # via jinja2
 
174
  matplotlib==3.9.1
175
  # via gradio
176
  mdurl==0.1.2
177
  # via markdown-it-py
 
 
 
178
  modelscope==1.16.1
179
  # via aitoolkits-webui
180
  mpmath==1.3.0
@@ -184,6 +212,8 @@ multidict==6.0.5
184
  # via yarl
185
  multiprocess==0.70.16
186
  # via datasets
 
 
187
  networkx==3.3
188
  # via torch
189
  numpy==1.26.4
@@ -192,17 +222,27 @@ numpy==1.26.4
192
  # via datasets
193
  # via diffusers
194
  # via gradio
 
 
195
  # via langchain
196
  # via matplotlib
 
197
  # via opencv-python
 
198
  # via pandas
199
  # via pyarrow
200
  # via scipy
 
 
201
  # via torchvision
202
  openai==1.35.13
203
  # via langchain-openai
204
  opencv-python==4.10.0.84
205
  # via aitoolkits-webui
 
 
 
 
206
  orjson==3.10.6
207
  # via fastapi
208
  # via gradio
@@ -216,9 +256,11 @@ packaging==24.1
216
  # via gradio
217
  # via gradio-client
218
  # via huggingface-hub
 
219
  # via kornia
220
  # via langchain-core
221
  # via matplotlib
 
222
  pandas==2.2.2
223
  # via altair
224
  # via datasets
@@ -228,6 +270,9 @@ pillow==10.4.0
228
  # via gradio
229
  # via matplotlib
230
  # via torchvision
 
 
 
231
  pyarrow==17.0.0
232
  # via datasets
233
  pyarrow-hotfix==0.6
@@ -289,8 +334,10 @@ requests==2.32.3
289
  # via langsmith
290
  # via modelscope
291
  # via oss2
 
292
  # via tiktoken
293
  rich==13.7.1
 
294
  # via typer
295
  rpds-py==0.19.0
296
  # via jsonschema
@@ -306,13 +353,19 @@ semantic-version==2.10.0
306
  setuptools==69.5.1
307
  # via aitoolkits-webui
308
  # via modelscope
 
 
309
  shellingham==1.5.4
310
  # via typer
311
  simplejson==3.19.2
312
  # via modelscope
313
  six==1.16.0
 
 
314
  # via oss2
315
  # via python-dateutil
 
 
316
  sniffio==1.3.1
317
  # via anyio
318
  # via httpx
@@ -328,6 +381,16 @@ sympy==1.13.0
328
  tenacity==8.5.0
329
  # via langchain
330
  # via langchain-core
 
 
 
 
 
 
 
 
 
 
331
  tiktoken==0.7.0
332
  # via langchain-openai
333
  tomlkit==0.12.0
@@ -359,9 +422,11 @@ typing-extensions==4.12.2
359
  # via gradio-client
360
  # via huggingface-hub
361
  # via openai
 
362
  # via pydantic
363
  # via pydantic-core
364
  # via sqlalchemy
 
365
  # via torch
366
  # via typer
367
  # via uvicorn
@@ -385,6 +450,12 @@ websocket-client==1.8.0
385
  websockets==11.0.3
386
  # via gradio-client
387
  # via uvicorn
 
 
 
 
 
 
388
  xxhash==3.4.1
389
  # via datasets
390
  yarl==1.9.4
 
9
  # generate-hashes: false
10
 
11
  -e file:.
12
+ absl-py==2.1.0
13
+ # via keras
14
+ # via tensorboard
15
+ # via tensorflow
16
  addict==2.4.0
17
  # via modelscope
18
  aiofiles==23.2.1
 
38
  # via openai
39
  # via starlette
40
  # via watchfiles
41
+ astunparse==1.6.3
42
+ # via tensorflow
43
  async-timeout==4.0.3
44
  # via aiohttp
45
  # via langchain
 
97
  # via diffusers
98
  # via huggingface-hub
99
  # via torch
100
+ flatbuffers==24.3.25
101
+ # via tensorflow
102
  fonttools==4.53.1
103
  # via matplotlib
104
  frozenlist==1.4.1
 
109
  # via gradio-client
110
  # via huggingface-hub
111
  # via torch
112
+ gast==0.6.0
113
+ # via tensorflow
114
+ google-pasta==0.2.0
115
+ # via tensorflow
116
  gradio==4.38.1
117
  # via aitoolkits-webui
118
  gradio-client==1.1.0
119
  # via gradio
120
+ grpcio==1.65.1
121
+ # via tensorboard
122
+ # via tensorflow
123
  h11==0.14.0
124
  # via httpcore
125
  # via uvicorn
126
+ h5py==3.11.0
127
+ # via keras
128
+ # via tensorflow
129
  httpcore==1.0.5
130
  # via httpx
131
  httptools==0.6.1
 
165
  # via altair
166
  jsonschema-specifications==2023.12.1
167
  # via jsonschema
168
+ keras==3.4.1
169
+ # via tensorflow
170
  kiwisolver==1.4.5
171
  # via matplotlib
172
  kornia==0.7.3
 
186
  langsmith==0.1.85
187
  # via langchain
188
  # via langchain-core
189
+ libclang==18.1.1
190
+ # via tensorflow
191
+ markdown==3.6
192
+ # via tensorboard
193
  markdown-it-py==3.0.0
194
  # via rich
195
  markupsafe==2.1.5
196
  # via gradio
197
  # via jinja2
198
+ # via werkzeug
199
  matplotlib==3.9.1
200
  # via gradio
201
  mdurl==0.1.2
202
  # via markdown-it-py
203
+ ml-dtypes==0.4.0
204
+ # via keras
205
+ # via tensorflow
206
  modelscope==1.16.1
207
  # via aitoolkits-webui
208
  mpmath==1.3.0
 
212
  # via yarl
213
  multiprocess==0.70.16
214
  # via datasets
215
+ namex==0.0.8
216
+ # via keras
217
  networkx==3.3
218
  # via torch
219
  numpy==1.26.4
 
222
  # via datasets
223
  # via diffusers
224
  # via gradio
225
+ # via h5py
226
+ # via keras
227
  # via langchain
228
  # via matplotlib
229
+ # via ml-dtypes
230
  # via opencv-python
231
+ # via opt-einsum
232
  # via pandas
233
  # via pyarrow
234
  # via scipy
235
+ # via tensorboard
236
+ # via tensorflow
237
  # via torchvision
238
  openai==1.35.13
239
  # via langchain-openai
240
  opencv-python==4.10.0.84
241
  # via aitoolkits-webui
242
+ opt-einsum==3.3.0
243
+ # via tensorflow
244
+ optree==0.12.1
245
+ # via keras
246
  orjson==3.10.6
247
  # via fastapi
248
  # via gradio
 
256
  # via gradio
257
  # via gradio-client
258
  # via huggingface-hub
259
+ # via keras
260
  # via kornia
261
  # via langchain-core
262
  # via matplotlib
263
+ # via tensorflow
264
  pandas==2.2.2
265
  # via altair
266
  # via datasets
 
270
  # via gradio
271
  # via matplotlib
272
  # via torchvision
273
+ protobuf==4.25.3
274
+ # via tensorboard
275
+ # via tensorflow
276
  pyarrow==17.0.0
277
  # via datasets
278
  pyarrow-hotfix==0.6
 
334
  # via langsmith
335
  # via modelscope
336
  # via oss2
337
+ # via tensorflow
338
  # via tiktoken
339
  rich==13.7.1
340
+ # via keras
341
  # via typer
342
  rpds-py==0.19.0
343
  # via jsonschema
 
353
  setuptools==69.5.1
354
  # via aitoolkits-webui
355
  # via modelscope
356
+ # via tensorboard
357
+ # via tensorflow
358
  shellingham==1.5.4
359
  # via typer
360
  simplejson==3.19.2
361
  # via modelscope
362
  six==1.16.0
363
+ # via astunparse
364
+ # via google-pasta
365
  # via oss2
366
  # via python-dateutil
367
+ # via tensorboard
368
+ # via tensorflow
369
  sniffio==1.3.1
370
  # via anyio
371
  # via httpx
 
381
  tenacity==8.5.0
382
  # via langchain
383
  # via langchain-core
384
+ tensorboard==2.17.0
385
+ # via tensorflow
386
+ tensorboard-data-server==0.7.2
387
+ # via tensorboard
388
+ tensorflow==2.17.0
389
+ # via aitoolkits-webui
390
+ tensorflow-io-gcs-filesystem==0.37.1
391
+ # via tensorflow
392
+ termcolor==2.4.0
393
+ # via tensorflow
394
  tiktoken==0.7.0
395
  # via langchain-openai
396
  tomlkit==0.12.0
 
422
  # via gradio-client
423
  # via huggingface-hub
424
  # via openai
425
+ # via optree
426
  # via pydantic
427
  # via pydantic-core
428
  # via sqlalchemy
429
+ # via tensorflow
430
  # via torch
431
  # via typer
432
  # via uvicorn
 
450
  websockets==11.0.3
451
  # via gradio-client
452
  # via uvicorn
453
+ werkzeug==3.0.3
454
+ # via tensorboard
455
+ wheel==0.43.0
456
+ # via astunparse
457
+ wrapt==1.16.0
458
+ # via tensorflow
459
  xxhash==3.4.1
460
  # via datasets
461
  yarl==1.9.4
requirements.txt CHANGED
@@ -8,6 +8,10 @@
8
  # with-sources: false
9
  # generate-hashes: false
10
 
 
 
 
 
11
  addict==2.4.0
12
  # via modelscope
13
  aiofiles==23.2.1
@@ -33,6 +37,8 @@ anyio==4.4.0
33
  # via openai
34
  # via starlette
35
  # via watchfiles
 
 
36
  async-timeout==4.0.3
37
  # via aiohttp
38
  # via langchain
@@ -90,6 +96,8 @@ filelock==3.15.4
90
  # via diffusers
91
  # via huggingface-hub
92
  # via torch
 
 
93
  fonttools==4.53.1
94
  # via matplotlib
95
  frozenlist==1.4.1
@@ -100,13 +108,23 @@ fsspec==2024.2.0
100
  # via gradio-client
101
  # via huggingface-hub
102
  # via torch
 
 
 
 
103
  gradio==4.38.1
104
  # via aitoolkits-webui
105
  gradio-client==1.1.0
106
  # via gradio
 
 
 
107
  h11==0.14.0
108
  # via httpcore
109
  # via uvicorn
 
 
 
110
  httpcore==1.0.5
111
  # via httpx
112
  httptools==0.6.1
@@ -146,6 +164,8 @@ jsonschema==4.23.0
146
  # via altair
147
  jsonschema-specifications==2023.12.1
148
  # via jsonschema
 
 
149
  kiwisolver==1.4.5
150
  # via matplotlib
151
  kornia==0.7.3
@@ -165,15 +185,23 @@ langchain-text-splitters==0.2.2
165
  langsmith==0.1.85
166
  # via langchain
167
  # via langchain-core
 
 
 
 
168
  markdown-it-py==3.0.0
169
  # via rich
170
  markupsafe==2.1.5
171
  # via gradio
172
  # via jinja2
 
173
  matplotlib==3.9.1
174
  # via gradio
175
  mdurl==0.1.2
176
  # via markdown-it-py
 
 
 
177
  modelscope==1.16.1
178
  # via aitoolkits-webui
179
  mpmath==1.3.0
@@ -183,6 +211,8 @@ multidict==6.0.5
183
  # via yarl
184
  multiprocess==0.70.16
185
  # via datasets
 
 
186
  networkx==3.3
187
  # via torch
188
  numpy==1.26.4
@@ -191,17 +221,27 @@ numpy==1.26.4
191
  # via datasets
192
  # via diffusers
193
  # via gradio
 
 
194
  # via langchain
195
  # via matplotlib
 
196
  # via opencv-python
 
197
  # via pandas
198
  # via pyarrow
199
  # via scipy
 
 
200
  # via torchvision
201
  openai==1.35.13
202
  # via langchain-openai
203
  opencv-python==4.10.0.84
204
  # via aitoolkits-webui
 
 
 
 
205
  orjson==3.10.6
206
  # via fastapi
207
  # via gradio
@@ -215,9 +255,11 @@ packaging==24.1
215
  # via gradio
216
  # via gradio-client
217
  # via huggingface-hub
 
218
  # via kornia
219
  # via langchain-core
220
  # via matplotlib
 
221
  pandas==2.2.2
222
  # via altair
223
  # via datasets
@@ -227,6 +269,9 @@ pillow==10.4.0
227
  # via gradio
228
  # via matplotlib
229
  # via torchvision
 
 
 
230
  pyarrow==17.0.0
231
  # via datasets
232
  pyarrow-hotfix==0.6
@@ -288,8 +333,10 @@ requests==2.32.3
288
  # via langsmith
289
  # via modelscope
290
  # via oss2
 
291
  # via tiktoken
292
  rich==13.7.1
 
293
  # via typer
294
  rpds-py==0.19.0
295
  # via jsonschema
@@ -305,13 +352,19 @@ semantic-version==2.10.0
305
  setuptools==69.5.1
306
  # via aitoolkits-webui
307
  # via modelscope
 
 
308
  shellingham==1.5.4
309
  # via typer
310
  simplejson==3.19.2
311
  # via modelscope
312
  six==1.16.0
 
 
313
  # via oss2
314
  # via python-dateutil
 
 
315
  sniffio==1.3.1
316
  # via anyio
317
  # via httpx
@@ -327,6 +380,16 @@ sympy==1.13.0
327
  tenacity==8.5.0
328
  # via langchain
329
  # via langchain-core
 
 
 
 
 
 
 
 
 
 
330
  tiktoken==0.7.0
331
  # via langchain-openai
332
  tomlkit==0.12.0
@@ -358,9 +421,11 @@ typing-extensions==4.12.2
358
  # via gradio-client
359
  # via huggingface-hub
360
  # via openai
 
361
  # via pydantic
362
  # via pydantic-core
363
  # via sqlalchemy
 
364
  # via torch
365
  # via typer
366
  # via uvicorn
@@ -384,6 +449,12 @@ websocket-client==1.8.0
384
  websockets==11.0.3
385
  # via gradio-client
386
  # via uvicorn
 
 
 
 
 
 
387
  xxhash==3.4.1
388
  # via datasets
389
  yarl==1.9.4
 
8
  # with-sources: false
9
  # generate-hashes: false
10
 
11
+ absl-py==2.1.0
12
+ # via keras
13
+ # via tensorboard
14
+ # via tensorflow
15
  addict==2.4.0
16
  # via modelscope
17
  aiofiles==23.2.1
 
37
  # via openai
38
  # via starlette
39
  # via watchfiles
40
+ astunparse==1.6.3
41
+ # via tensorflow
42
  async-timeout==4.0.3
43
  # via aiohttp
44
  # via langchain
 
96
  # via diffusers
97
  # via huggingface-hub
98
  # via torch
99
+ flatbuffers==24.3.25
100
+ # via tensorflow
101
  fonttools==4.53.1
102
  # via matplotlib
103
  frozenlist==1.4.1
 
108
  # via gradio-client
109
  # via huggingface-hub
110
  # via torch
111
+ gast==0.6.0
112
+ # via tensorflow
113
+ google-pasta==0.2.0
114
+ # via tensorflow
115
  gradio==4.38.1
116
  # via aitoolkits-webui
117
  gradio-client==1.1.0
118
  # via gradio
119
+ grpcio==1.65.1
120
+ # via tensorboard
121
+ # via tensorflow
122
  h11==0.14.0
123
  # via httpcore
124
  # via uvicorn
125
+ h5py==3.11.0
126
+ # via keras
127
+ # via tensorflow
128
  httpcore==1.0.5
129
  # via httpx
130
  httptools==0.6.1
 
164
  # via altair
165
  jsonschema-specifications==2023.12.1
166
  # via jsonschema
167
+ keras==3.4.1
168
+ # via tensorflow
169
  kiwisolver==1.4.5
170
  # via matplotlib
171
  kornia==0.7.3
 
185
  langsmith==0.1.85
186
  # via langchain
187
  # via langchain-core
188
+ libclang==18.1.1
189
+ # via tensorflow
190
+ markdown==3.6
191
+ # via tensorboard
192
  markdown-it-py==3.0.0
193
  # via rich
194
  markupsafe==2.1.5
195
  # via gradio
196
  # via jinja2
197
+ # via werkzeug
198
  matplotlib==3.9.1
199
  # via gradio
200
  mdurl==0.1.2
201
  # via markdown-it-py
202
+ ml-dtypes==0.4.0
203
+ # via keras
204
+ # via tensorflow
205
  modelscope==1.16.1
206
  # via aitoolkits-webui
207
  mpmath==1.3.0
 
211
  # via yarl
212
  multiprocess==0.70.16
213
  # via datasets
214
+ namex==0.0.8
215
+ # via keras
216
  networkx==3.3
217
  # via torch
218
  numpy==1.26.4
 
221
  # via datasets
222
  # via diffusers
223
  # via gradio
224
+ # via h5py
225
+ # via keras
226
  # via langchain
227
  # via matplotlib
228
+ # via ml-dtypes
229
  # via opencv-python
230
+ # via opt-einsum
231
  # via pandas
232
  # via pyarrow
233
  # via scipy
234
+ # via tensorboard
235
+ # via tensorflow
236
  # via torchvision
237
  openai==1.35.13
238
  # via langchain-openai
239
  opencv-python==4.10.0.84
240
  # via aitoolkits-webui
241
+ opt-einsum==3.3.0
242
+ # via tensorflow
243
+ optree==0.12.1
244
+ # via keras
245
  orjson==3.10.6
246
  # via fastapi
247
  # via gradio
 
255
  # via gradio
256
  # via gradio-client
257
  # via huggingface-hub
258
+ # via keras
259
  # via kornia
260
  # via langchain-core
261
  # via matplotlib
262
+ # via tensorflow
263
  pandas==2.2.2
264
  # via altair
265
  # via datasets
 
269
  # via gradio
270
  # via matplotlib
271
  # via torchvision
272
+ protobuf==4.25.3
273
+ # via tensorboard
274
+ # via tensorflow
275
  pyarrow==17.0.0
276
  # via datasets
277
  pyarrow-hotfix==0.6
 
333
  # via langsmith
334
  # via modelscope
335
  # via oss2
336
+ # via tensorflow
337
  # via tiktoken
338
  rich==13.7.1
339
+ # via keras
340
  # via typer
341
  rpds-py==0.19.0
342
  # via jsonschema
 
352
  setuptools==69.5.1
353
  # via aitoolkits-webui
354
  # via modelscope
355
+ # via tensorboard
356
+ # via tensorflow
357
  shellingham==1.5.4
358
  # via typer
359
  simplejson==3.19.2
360
  # via modelscope
361
  six==1.16.0
362
+ # via astunparse
363
+ # via google-pasta
364
  # via oss2
365
  # via python-dateutil
366
+ # via tensorboard
367
+ # via tensorflow
368
  sniffio==1.3.1
369
  # via anyio
370
  # via httpx
 
380
  tenacity==8.5.0
381
  # via langchain
382
  # via langchain-core
383
+ tensorboard==2.17.0
384
+ # via tensorflow
385
+ tensorboard-data-server==0.7.2
386
+ # via tensorboard
387
+ tensorflow==2.17.0
388
+ # via aitoolkits-webui
389
+ tensorflow-io-gcs-filesystem==0.37.1
390
+ # via tensorflow
391
+ termcolor==2.4.0
392
+ # via tensorflow
393
  tiktoken==0.7.0
394
  # via langchain-openai
395
  tomlkit==0.12.0
 
421
  # via gradio-client
422
  # via huggingface-hub
423
  # via openai
424
+ # via optree
425
  # via pydantic
426
  # via pydantic-core
427
  # via sqlalchemy
428
+ # via tensorflow
429
  # via torch
430
  # via typer
431
  # via uvicorn
 
449
  websockets==11.0.3
450
  # via gradio-client
451
  # via uvicorn
452
+ werkzeug==3.0.3
453
+ # via tensorboard
454
+ wheel==0.43.0
455
+ # via astunparse
456
+ wrapt==1.16.0
457
+ # via tensorflow
458
  xxhash==3.4.1
459
  # via datasets
460
  yarl==1.9.4