Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/basketball.mp4 filter=lfs diff=lfs merge=lfs -text
+assets/pierre.mov filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,3 @@
+.venv
+node_modules
+tests

README.md

@@ -1,12 +1,6 @@
 ---
-title: Human Pose Video
-emoji: 🐢
-colorFrom: green
-colorTo: blue
+title: human-pose-video
+app_file: app.py
 sdk: gradio
 sdk_version: 4.42.0
-app_file: app.py
-pinned: false
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+import PIL.Image as Image
+from ultralytics import YOLO
+
+pose = YOLO("models/pose.pt")
+
+
+def predict_image(image, conf_threshold, iou_threshold):
+    results = pose.predict(
+        image, conf=conf_threshold, iou=iou_threshold, stream=True)
+
+    for r in results:
+        im_array = r.plot(labels=True, boxes=True)
+        yield Image.fromarray(im_array[..., ::-1])
+
+    return image
+
+
+iface = gr.Interface(
+    fn=predict_image,
+    inputs=[
+        gr.Video(label="Upload Video"),
+        gr.Slider(minimum=0, maximum=1, value=0.25,
+                  label="Confidence threshold"),
+        gr.Slider(minimum=0, maximum=1, value=0.7, label="IoU threshold"),
+    ],
+    outputs=gr.Image(type="numpy", label="Result"),
+    title="Human Pose",
+    description="Limbs in all of the right places.",
+    examples=[
+        ["assets/pierre.mov", 0.25, 0.7],
+        ["assets/basketball.mp4", 0.25, 0.7],
+    ]
+)
+
+if __name__ == "__main__":
+    iface.launch()

assets/basketball.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1996f65f6335c4bbdf436a992ee3f7d9db7a60bbd113f7305c53e572c7260759
+size 30162927

assets/pierre.mov

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64e020d6ad93a93c0134b1e881941ca14c16bcddade5b795e0ef155554d00b61
+size 51297175

models/pose.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a14086802581046de9b39939924440523252cd5a47361081e802984ddf714ea1
+size 89391502

requirements.txt

@@ -0,0 +1,69 @@
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.4.0
+certifi==2024.7.4
+charset-normalizer==3.3.2
+click==8.1.7
+contourpy==1.3.0
+cycler==0.12.1
+fastapi==0.112.2
+ffmpy==0.4.0
+filelock==3.15.4
+fonttools==4.53.1
+fsspec==2024.6.1
+gradio==4.42.0
+gradio_client==1.3.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.2
+huggingface-hub==0.24.6
+idna==3.8
+importlib_resources==6.4.4
+Jinja2==3.1.4
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.9.2
+mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.3
+numpy==1.26.4
+opencv-python==4.10.0.84
+orjson==3.10.7
+packaging==24.1
+pandas==2.2.2
+pillow==10.4.0
+psutil==6.0.0
+py-cpuinfo==9.0.0
+pydantic==2.8.2
+pydantic_core==2.20.1
+pydub==0.25.1
+Pygments==2.18.0
+pyparsing==3.1.4
+python-dateutil==2.9.0.post0
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.2
+requests==2.32.3
+rich==13.8.0
+ruff==0.6.2
+scipy==1.14.1
+seaborn==0.13.2
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.38.2
+sympy==1.13.2
+tomlkit==0.12.0
+torch==2.4.0
+torchvision==0.19.0
+tqdm==4.66.5
+typer==0.12.5
+typing_extensions==4.12.2
+tzdata==2024.1
+ultralytics==8.2.83
+ultralytics-thop==2.0.5
+urllib3==2.2.2
+uvicorn==0.30.6
+websockets==12.0