Update detect_bubbles.py

detect_bubbles.py

@@ -6,7 +6,7 @@ def safe_load_model(model_path):
     Manually loads the YOLO model weights with 'weights_only' set to False
     to avoid the security check issue with PyTorch.
     """
-    model = torch.load(model_path, map_location='cpu', weights_only=False)
+    model = YOLO(model_path)
     return model
 
 def detect_bubbles(model_path, image_path):
@@ -19,7 +19,8 @@ def detect_bubbles(model_path, image_path):
         list: A list containing the coordinates, score, and class_id of 
               the detected bubbles.
     """
-    model = safe_load_model(model_path)
-    bubbles = model(image_path)[0]
+    model = safe_load_model(model_path) 
+    results = model(image_path)
+    bubbles = results[0].boxes.data.tolist()
 
-    return bubbles.boxes.data.tolist()
+    return bubbles