mrisdi commited on
Commit
f626de0
·
1 Parent(s): 7bb4355

feat: gradio

Browse files
Files changed (4) hide show
  1. .gitignore +2 -0
  2. app.py +17 -0
  3. requirements.txt +83 -0
  4. yolo11n-pose.pt +3 -0
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ venv
2
+ .gradio
app.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from ultralytics import YOLO
3
+
4
+ model = YOLO('yolo11n-pose.pt')
5
+
6
+ def poseImage(image):
7
+ results = model(image)
8
+ return results[0].plot()
9
+
10
+ interface = gr.Interface(
11
+ fn=poseImage,
12
+ inputs=gr.Image(streaming=True),
13
+ outputs=gr.Image(),
14
+ live=True
15
+ )
16
+
17
+ interface.launch()
requirements.txt ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ aiofiles==23.2.1
2
+ aioice==0.9.0
3
+ aiortc==1.9.0
4
+ annotated-types==0.7.0
5
+ anyio==4.6.2.post1
6
+ av==12.3.0
7
+ certifi==2024.8.30
8
+ cffi==1.17.1
9
+ charset-normalizer==3.4.0
10
+ click==8.1.7
11
+ colorama==0.4.6
12
+ contourpy==1.3.1
13
+ cryptography==43.0.3
14
+ cycler==0.12.1
15
+ dnspython==2.7.0
16
+ exceptiongroup==1.2.2
17
+ fastapi==0.115.5
18
+ ffmpy==0.4.0
19
+ filelock==3.16.1
20
+ fonttools==4.55.0
21
+ fsspec==2024.10.0
22
+ google-crc32c==1.6.0
23
+ gradio==5.6.0
24
+ gradio_client==1.4.3
25
+ h11==0.14.0
26
+ httpcore==1.0.7
27
+ httpx==0.27.2
28
+ huggingface-hub==0.26.2
29
+ idna==3.10
30
+ ifaddr==0.2.0
31
+ Jinja2==3.1.4
32
+ kiwisolver==1.4.7
33
+ markdown-it-py==3.0.0
34
+ MarkupSafe==2.1.5
35
+ matplotlib==3.9.2
36
+ mdurl==0.1.2
37
+ mpmath==1.3.0
38
+ networkx==3.4.2
39
+ numpy==2.1.3
40
+ opencv-python==4.10.0.84
41
+ orjson==3.10.11
42
+ packaging==24.2
43
+ pandas==2.2.3
44
+ pillow==11.0.0
45
+ psutil==6.1.0
46
+ py-cpuinfo==9.0.0
47
+ pycparser==2.22
48
+ pydantic==2.9.2
49
+ pydantic_core==2.23.4
50
+ pydub==0.25.1
51
+ pyee==12.1.1
52
+ Pygments==2.18.0
53
+ pylibsrtp==0.10.0
54
+ pyOpenSSL==24.2.1
55
+ pyparsing==3.2.0
56
+ python-dateutil==2.9.0.post0
57
+ python-multipart==0.0.12
58
+ pytz==2024.2
59
+ PyYAML==6.0.2
60
+ requests==2.32.3
61
+ rich==13.9.4
62
+ ruff==0.7.4
63
+ safehttpx==0.1.1
64
+ scipy==1.14.1
65
+ seaborn==0.13.2
66
+ semantic-version==2.10.0
67
+ shellingham==1.5.4
68
+ six==1.16.0
69
+ sniffio==1.3.1
70
+ starlette==0.41.2
71
+ sympy==1.13.1
72
+ tomlkit==0.12.0
73
+ torch==2.5.1
74
+ torchvision==0.20.1
75
+ tqdm==4.67.0
76
+ typer==0.13.0
77
+ typing_extensions==4.12.2
78
+ tzdata==2024.2
79
+ ultralytics==8.3.32
80
+ ultralytics-thop==2.0.11
81
+ urllib3==2.2.3
82
+ uvicorn==0.32.0
83
+ websockets==12.0
yolo11n-pose.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:869e83fcdffdc7371fa4e34cd8e51c838cc729571d1635e5141e3075e9319dc0
3
+ size 6255593