Update README.md

README.md

@@ -73,4 +73,69 @@ plt.imshow(plt.imread('detected/sample.jpg'))
 plt.show()
 ```
 
-As you can see the model isn't perfect ;)
+As you can see the model isn't perfect ;)
+
+
+### Use the model with your webcam
+
+```
+from ultralytics import YOLO
+import cv2
+import math 
+from huggingface_hub import hf_hub_download
+
+# start the webcam
+cap = cv2.VideoCapture(0)
+cap.set(3, 640)
+cap.set(4, 480)
+
+# Load the weights from our repository
+model_path = hf_hub_download(
+    local_dir=".",
+    repo_id="fairportrobotics/rock-paper-scissors",
+    filename="model.pt"
+)
+model = YOLO(model_path)
+
+# object classes
+classNames = ["rock", "paper", "scissors"]
+
+
+while True:
+    success, img = cap.read()
+    results = model(img, stream=True)
+
+    # coordinates
+    for r in results:
+        boxes = r.boxes
+
+        for box in boxes:
+            # bounding box
+            x1, y1, x2, y2 = box.xyxy[0]
+            x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2) # convert to int values
+
+            # put box in cam
+            cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 255), 3)
+
+            # confidence
+            confidence = math.ceil((box.conf[0]*100))/100
+
+            # class name
+            cls = int(box.cls[0])
+
+            # object details
+            org = [x1, y1]
+            font = cv2.FONT_HERSHEY_SIMPLEX
+            fontScale = 1
+            color = (255, 0, 0)
+            thickness = 2
+
+            cv2.putText(img, classNames[cls] + " " + str(round(confidence,2)), org, font, fontScale, color, thickness)
+
+    cv2.imshow('Webcam', img)
+    if cv2.waitKey(1) == ord('q'):
+        break
+
+cap.release()
+cv2.destroyAllWindows()
+```