hysts HF Staff commited on
Commit
b8aa7b5
·
1 Parent(s): b31249e
Files changed (6) hide show
  1. .pre-commit-config.yaml +3 -3
  2. README.md +3 -2
  3. app.py +1 -7
  4. pyproject.toml +8 -10
  5. requirements.txt +26 -24
  6. uv.lock +0 -0
.pre-commit-config.yaml CHANGED
@@ -14,13 +14,13 @@ repos:
14
  - id: requirements-txt-fixer
15
  - id: trailing-whitespace
16
  - repo: https://github.com/astral-sh/ruff-pre-commit
17
- rev: v0.8.6
18
  hooks:
19
- - id: ruff
20
  args: ["--fix"]
21
  - id: ruff-format
22
  - repo: https://github.com/pre-commit/mirrors-mypy
23
- rev: v1.14.1
24
  hooks:
25
  - id: mypy
26
  args: ["--ignore-missing-imports"]
 
14
  - id: requirements-txt-fixer
15
  - id: trailing-whitespace
16
  - repo: https://github.com/astral-sh/ruff-pre-commit
17
+ rev: v0.12.1
18
  hooks:
19
+ - id: ruff-check
20
  args: ["--fix"]
21
  - id: ruff-format
22
  - repo: https://github.com/pre-commit/mirrors-mypy
23
+ rev: v1.16.1
24
  hooks:
25
  - id: mypy
26
  args: ["--ignore-missing-imports"]
README.md CHANGED
@@ -4,9 +4,10 @@ emoji: ⚡
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 5.13.2
8
  app_file: app.py
9
  pinned: false
10
  ---
11
 
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 5.35.0
8
  app_file: app.py
9
  pinned: false
10
  ---
11
 
12
+ This code is based on the implementation from the Colab notebook:
13
+ https://colab.research.google.com/drive/1e8fcby5rhKZWcr9LSN8mNbQ0TU4Dxxpo
app.py CHANGED
@@ -1,11 +1,5 @@
1
  #!/usr/bin/env python
2
 
3
- """A demo of the VitPose model.
4
-
5
- This code is based on the implementation from the Colab notebook:
6
- https://colab.research.google.com/drive/1e8fcby5rhKZWcr9LSN8mNbQ0TU4Dxxpo
7
- """
8
-
9
  import pathlib
10
  import tempfile
11
 
@@ -184,4 +178,4 @@ with gr.Blocks(css_paths="style.css") as demo:
184
 
185
 
186
  if __name__ == "__main__":
187
- demo.queue(max_size=20).launch()
 
1
  #!/usr/bin/env python
2
 
 
 
 
 
 
 
3
  import pathlib
4
  import tempfile
5
 
 
178
 
179
 
180
  if __name__ == "__main__":
181
+ demo.launch()
pyproject.toml CHANGED
@@ -5,15 +5,14 @@ description = ""
5
  readme = "README.md"
6
  requires-python = ">=3.10"
7
  dependencies = [
8
- "accelerate>=1.3.0",
9
- "gradio>=5.13.2",
10
  "hf-transfer>=0.1.9",
11
- "opencv-python-headless>=4.11.0.86",
12
- "setuptools>=75.8.0",
13
- "spaces>=0.32.0",
14
  "supervision>=0.25.1",
15
- "torch==2.4.0",
16
- "transformers>=4.48.1",
17
  ]
18
 
19
  [tool.ruff]
@@ -27,8 +26,7 @@ ignore = [
27
  "D213", # multi-line-summary-second-line
28
  "E501", # line-too-long
29
  "SIM117", # multiple-with-statements
30
- ]
31
- extend-ignore = [
32
  "D100", # undocumented-public-module
33
  "D101", # undocumented-public-class
34
  "D102", # undocumented-public-method
@@ -53,7 +51,7 @@ unfixable = [
53
  convention = "google"
54
 
55
  [tool.ruff.lint.per-file-ignores]
56
- "*.ipynb" = ["T201"]
57
 
58
  [tool.ruff.format]
59
  docstring-code-format = true
 
5
  readme = "README.md"
6
  requires-python = ">=3.10"
7
  dependencies = [
8
+ "accelerate>=1.8.1",
9
+ "gradio>=5.35.0",
10
  "hf-transfer>=0.1.9",
11
+ "setuptools>=80.9.0",
12
+ "spaces>=0.37.1",
 
13
  "supervision>=0.25.1",
14
+ "torch==2.5.1",
15
+ "transformers>=4.53.0",
16
  ]
17
 
18
  [tool.ruff]
 
26
  "D213", # multi-line-summary-second-line
27
  "E501", # line-too-long
28
  "SIM117", # multiple-with-statements
29
+ #
 
30
  "D100", # undocumented-public-module
31
  "D101", # undocumented-public-class
32
  "D102", # undocumented-public-method
 
51
  convention = "google"
52
 
53
  [tool.ruff.lint.per-file-ignores]
54
+ "*.ipynb" = ["T201", "T203"]
55
 
56
  [tool.ruff.format]
57
  docstring-code-format = true
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  # This file was autogenerated by uv via the following command:
2
  # uv pip compile pyproject.toml -o requirements.txt
3
- accelerate==1.3.0
4
  # via vitpose-transformers (pyproject.toml)
5
  aiofiles==23.2.1
6
  # via gradio
@@ -49,11 +49,13 @@ fsspec==2024.12.0
49
  # gradio-client
50
  # huggingface-hub
51
  # torch
52
- gradio==5.13.2
53
  # via
54
  # vitpose-transformers (pyproject.toml)
55
  # spaces
56
- gradio-client==1.6.0
 
 
57
  # via gradio
58
  h11==0.14.0
59
  # via
@@ -61,6 +63,8 @@ h11==0.14.0
61
  # uvicorn
62
  hf-transfer==0.1.9
63
  # via vitpose-transformers (pyproject.toml)
 
 
64
  httpcore==1.0.7
65
  # via httpx
66
  httpx==0.28.1
@@ -69,7 +73,7 @@ httpx==0.28.1
69
  # gradio-client
70
  # safehttpx
71
  # spaces
72
- huggingface-hub==0.28.0
73
  # via
74
  # accelerate
75
  # gradio
@@ -108,46 +112,44 @@ numpy==2.2.2
108
  # gradio
109
  # matplotlib
110
  # opencv-python
111
- # opencv-python-headless
112
  # pandas
113
  # scipy
114
  # supervision
115
  # transformers
116
- nvidia-cublas-cu12==12.1.3.1
117
  # via
118
  # nvidia-cudnn-cu12
119
  # nvidia-cusolver-cu12
120
  # torch
121
- nvidia-cuda-cupti-cu12==12.1.105
122
  # via torch
123
- nvidia-cuda-nvrtc-cu12==12.1.105
124
  # via torch
125
- nvidia-cuda-runtime-cu12==12.1.105
126
  # via torch
127
  nvidia-cudnn-cu12==9.1.0.70
128
  # via torch
129
- nvidia-cufft-cu12==11.0.2.54
130
  # via torch
131
- nvidia-curand-cu12==10.3.2.106
132
  # via torch
133
- nvidia-cusolver-cu12==11.4.5.107
134
  # via torch
135
- nvidia-cusparse-cu12==12.1.0.106
136
  # via
137
  # nvidia-cusolver-cu12
138
  # torch
139
- nvidia-nccl-cu12==2.20.5
140
  # via torch
141
- nvidia-nvjitlink-cu12==12.8.61
142
  # via
143
  # nvidia-cusolver-cu12
144
  # nvidia-cusparse-cu12
145
- nvidia-nvtx-cu12==12.1.105
 
146
  # via torch
147
  opencv-python==4.11.0.86
148
  # via supervision
149
- opencv-python-headless==4.11.0.86
150
- # via vitpose-transformers (pyproject.toml)
151
  orjson==3.10.15
152
  # via gradio
153
  packaging==24.2
@@ -220,7 +222,7 @@ scipy==1.15.1
220
  # via supervision
221
  semantic-version==2.10.0
222
  # via gradio
223
- setuptools==75.8.0
224
  # via vitpose-transformers (pyproject.toml)
225
  shellingham==1.5.4
226
  # via typer
@@ -228,7 +230,7 @@ six==1.17.0
228
  # via python-dateutil
229
  sniffio==1.3.1
230
  # via anyio
231
- spaces==0.32.0
232
  # via vitpose-transformers (pyproject.toml)
233
  starlette==0.45.3
234
  # via
@@ -236,13 +238,13 @@ starlette==0.45.3
236
  # gradio
237
  supervision==0.25.1
238
  # via vitpose-transformers (pyproject.toml)
239
- sympy==1.13.3
240
  # via torch
241
  tokenizers==0.21.0
242
  # via transformers
243
  tomlkit==0.13.2
244
  # via gradio
245
- torch==2.4.0
246
  # via
247
  # vitpose-transformers (pyproject.toml)
248
  # accelerate
@@ -251,9 +253,9 @@ tqdm==4.67.1
251
  # huggingface-hub
252
  # supervision
253
  # transformers
254
- transformers==4.48.1
255
  # via vitpose-transformers (pyproject.toml)
256
- triton==3.0.0
257
  # via torch
258
  typer==0.15.1
259
  # via gradio
 
1
  # This file was autogenerated by uv via the following command:
2
  # uv pip compile pyproject.toml -o requirements.txt
3
+ accelerate==1.8.1
4
  # via vitpose-transformers (pyproject.toml)
5
  aiofiles==23.2.1
6
  # via gradio
 
49
  # gradio-client
50
  # huggingface-hub
51
  # torch
52
+ gradio==5.35.0
53
  # via
54
  # vitpose-transformers (pyproject.toml)
55
  # spaces
56
+ gradio-client==1.10.4
57
+ # via gradio
58
+ groovy==0.1.2
59
  # via gradio
60
  h11==0.14.0
61
  # via
 
63
  # uvicorn
64
  hf-transfer==0.1.9
65
  # via vitpose-transformers (pyproject.toml)
66
+ hf-xet==1.1.5
67
+ # via huggingface-hub
68
  httpcore==1.0.7
69
  # via httpx
70
  httpx==0.28.1
 
73
  # gradio-client
74
  # safehttpx
75
  # spaces
76
+ huggingface-hub==0.33.2
77
  # via
78
  # accelerate
79
  # gradio
 
112
  # gradio
113
  # matplotlib
114
  # opencv-python
 
115
  # pandas
116
  # scipy
117
  # supervision
118
  # transformers
119
+ nvidia-cublas-cu12==12.4.5.8
120
  # via
121
  # nvidia-cudnn-cu12
122
  # nvidia-cusolver-cu12
123
  # torch
124
+ nvidia-cuda-cupti-cu12==12.4.127
125
  # via torch
126
+ nvidia-cuda-nvrtc-cu12==12.4.127
127
  # via torch
128
+ nvidia-cuda-runtime-cu12==12.4.127
129
  # via torch
130
  nvidia-cudnn-cu12==9.1.0.70
131
  # via torch
132
+ nvidia-cufft-cu12==11.2.1.3
133
  # via torch
134
+ nvidia-curand-cu12==10.3.5.147
135
  # via torch
136
+ nvidia-cusolver-cu12==11.6.1.9
137
  # via torch
138
+ nvidia-cusparse-cu12==12.3.1.170
139
  # via
140
  # nvidia-cusolver-cu12
141
  # torch
142
+ nvidia-nccl-cu12==2.21.5
143
  # via torch
144
+ nvidia-nvjitlink-cu12==12.4.127
145
  # via
146
  # nvidia-cusolver-cu12
147
  # nvidia-cusparse-cu12
148
+ # torch
149
+ nvidia-nvtx-cu12==12.4.127
150
  # via torch
151
  opencv-python==4.11.0.86
152
  # via supervision
 
 
153
  orjson==3.10.15
154
  # via gradio
155
  packaging==24.2
 
222
  # via supervision
223
  semantic-version==2.10.0
224
  # via gradio
225
+ setuptools==80.9.0
226
  # via vitpose-transformers (pyproject.toml)
227
  shellingham==1.5.4
228
  # via typer
 
230
  # via python-dateutil
231
  sniffio==1.3.1
232
  # via anyio
233
+ spaces==0.37.1
234
  # via vitpose-transformers (pyproject.toml)
235
  starlette==0.45.3
236
  # via
 
238
  # gradio
239
  supervision==0.25.1
240
  # via vitpose-transformers (pyproject.toml)
241
+ sympy==1.13.1
242
  # via torch
243
  tokenizers==0.21.0
244
  # via transformers
245
  tomlkit==0.13.2
246
  # via gradio
247
+ torch==2.5.1
248
  # via
249
  # vitpose-transformers (pyproject.toml)
250
  # accelerate
 
253
  # huggingface-hub
254
  # supervision
255
  # transformers
256
+ transformers==4.53.0
257
  # via vitpose-transformers (pyproject.toml)
258
+ triton==3.1.0
259
  # via torch
260
  typer==0.15.1
261
  # via gradio
uv.lock CHANGED
The diff for this file is too large to render. See raw diff