Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ 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/pierre.png filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+env
+node_modules

README.md

@@ -1,12 +1,6 @@
 ---
-title: Basketball Classify
-emoji: 😻
-colorFrom: gray
-colorTo: gray
+title: basketball-classify
+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,40 @@
+import gradio as gr
+import PIL.Image as Image
+from ultralytics import YOLO
+
+classify = YOLO("models/classify.pt")
+
+
+def predict(model, image, conf_threshold, iou_threshold, show=True):
+    results = model.predict(image, conf=conf_threshold, iou=iou_threshold)
+
+    for r in results:
+        im_array = r.plot(labels=show, boxes=show)
+        image = Image.fromarray(im_array[..., ::-1])
+
+    return image
+
+
+def predict_image(image, conf_threshold, iou_threshold):
+    return predict(classify, image, conf_threshold, iou_threshold)
+
+
+iface = gr.Interface(
+    fn=predict_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        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="pil", label="Result"),
+    title="Basketball Classifier",
+    description="Have you ever wondered where the ball was when you were playing basketball? Where the rim was? Where you were?",
+    examples=[
+        ["assets/klay.jpeg", 0.25, 0.7],
+        ["assets/pierre.png", 0.25, 0.7],
+    ]
+)
+
+if __name__ == "__main__":
+    iface.launch()

models/classify.pt

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

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