From b22b547fdcb92432df6a1be5680512d78dfcb79d Mon Sep 17 00:00:00 2001
From: Siwat Sirichai <siwat@siwatinc.com>
Date: Tue, 14 Jan 2025 22:35:22 +0700
Subject: [PATCH] change tracking parameter

---
 app.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/app.py b/app.py
index 595a40e..180583b 100644
--- a/app.py
+++ b/app.py
@@ -24,8 +24,7 @@ if torch.cuda.is_available():
     model.to('cuda')
 model.track(
     persist=True,
-    tracker="bytetrack.yaml",  # You can choose a different tracker if desired
-    track_kps=False
+    tracker="bytetrack.yaml"  # You can choose a different tracker if desired
 )
 
 # Retrieve class names from the model
@@ -124,7 +123,7 @@ async def detect(websocket: WebSocket):
                     buffer = stream['buffer']
                     if not buffer.empty():
                         frame = buffer.get()
-                        results = model.track(frame, stream=False)
+                        results = model.track(frame, stream=False)  # Updated for tracking
                         boxes = []
                         for r in results:
                             for track in r.tracks:
@@ -135,7 +134,7 @@ async def detect(websocket: WebSocket):
                                 boxes.append({
                                     "class": class_names[cls],
                                     "confidence": float(track.conf),
-                                    "track_id": track_id
+                                    "track_id": track_id  # Added track ID
                                 })
                         # Broadcast to all subscribers of this URL
                         detection_data = {