Update detect_bubbles.py

detect_bubbles.py

@@ -1,12 +1,14 @@
 from ultralytics import YOLO
 import torch
 
+torch.serialization.add_safe_globals([YOLO, 'DetectionModel'])
+
 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 = YOLO(model_path)
+    model = YOLO(model_path, weights_only=False)
     return model
 
 def detect_bubbles(model_path, image_path):