Upload folder using huggingface_hub

.gitattributes

@@ -34,3 +34,5 @@ saved_model/**/* 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
+assets/basketball.mp4 filter=lfs diff=lfs merge=lfs -text
+assets/pierre.mov filter=lfs diff=lfs merge=lfs -text

.gitignore

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

app.py

@@ -5,34 +5,31 @@ 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)
+def predict_image(image, conf_threshold, iou_threshold):
+    results = classify.predict(
+        image, conf=conf_threshold, iou=iou_threshold, stream=True)
 
     for r in results:
-        im_array = r.plot(labels=show, boxes=show)
-        image = Image.fromarray(im_array[..., ::-1])
+        im_array = r.plot(labels=True, boxes=True)
+        yield 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.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="pil", label="Result"),
+    outputs=gr.Image(type="numpy", 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],
+        ["assets/pierre.mov", 0.25, 0.7],
+        ["assets/basketball.mp4", 0.25, 0.7],
     ]
 )
 

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