Update miner.py

miner.py

@@ -92,16 +92,13 @@ class Miner:
                 boxes = []
                 for box in detection.boxes.data:
                     x1, y1, x2, y2, conf, cls_id = box.tolist()
-                    class_id = int(cls_id)
-                    if class_id == 2:
-                        class_id = 7
                     boxes.append(
                         BoundingBox(
                             x1=int(x1),
                             y1=int(y1),
                             x2=int(x2),
                             y2=int(y2),
-                            cls_id=class_id,
+                            cls_id=int(cls_id),
                             conf=float(conf),
                         )
                     )