Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files- .pre-commit-config.yaml +3 -3
- README.md +3 -2
- app.py +1 -7
- pyproject.toml +8 -10
- requirements.txt +26 -24
- 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.
|
18 |
hooks:
|
19 |
-
- id: ruff
|
20 |
args: ["--fix"]
|
21 |
- id: ruff-format
|
22 |
- repo: https://github.com/pre-commit/mirrors-mypy
|
23 |
-
rev: v1.
|
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.
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
11 |
|
12 |
-
|
|
|
|
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.
|
|
|
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.
|
9 |
-
"gradio>=5.
|
10 |
"hf-transfer>=0.1.9",
|
11 |
-
"
|
12 |
-
"
|
13 |
-
"spaces>=0.32.0",
|
14 |
"supervision>=0.25.1",
|
15 |
-
"torch==2.
|
16 |
-
"transformers>=4.
|
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.
|
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.
|
53 |
# via
|
54 |
# vitpose-transformers (pyproject.toml)
|
55 |
# spaces
|
56 |
-
gradio-client==1.
|
|
|
|
|
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.
|
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.
|
117 |
# via
|
118 |
# nvidia-cudnn-cu12
|
119 |
# nvidia-cusolver-cu12
|
120 |
# torch
|
121 |
-
nvidia-cuda-cupti-cu12==12.
|
122 |
# via torch
|
123 |
-
nvidia-cuda-nvrtc-cu12==12.
|
124 |
# via torch
|
125 |
-
nvidia-cuda-runtime-cu12==12.
|
126 |
# via torch
|
127 |
nvidia-cudnn-cu12==9.1.0.70
|
128 |
# via torch
|
129 |
-
nvidia-cufft-cu12==11.
|
130 |
# via torch
|
131 |
-
nvidia-curand-cu12==10.3.
|
132 |
# via torch
|
133 |
-
nvidia-cusolver-cu12==11.
|
134 |
# via torch
|
135 |
-
nvidia-cusparse-cu12==12.1.
|
136 |
# via
|
137 |
# nvidia-cusolver-cu12
|
138 |
# torch
|
139 |
-
nvidia-nccl-cu12==2.
|
140 |
# via torch
|
141 |
-
nvidia-nvjitlink-cu12==12.
|
142 |
# via
|
143 |
# nvidia-cusolver-cu12
|
144 |
# nvidia-cusparse-cu12
|
145 |
-
|
|
|
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==
|
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.
|
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.
|
240 |
# via torch
|
241 |
tokenizers==0.21.0
|
242 |
# via transformers
|
243 |
tomlkit==0.13.2
|
244 |
# via gradio
|
245 |
-
torch==2.
|
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.
|
255 |
# via vitpose-transformers (pyproject.toml)
|
256 |
-
triton==3.
|
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
|
|