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
|
|
|