Add torch safe globals for model loading

Amp-Thread-ID: https://ampcode.com/threads/T-22e83726-d77c-475f-b185-6d55f37c5979
Co-authored-by: Amp <amp@ampcode.com>

model.py

@@ -4,6 +4,7 @@ from ultralytics import YOLO
 import yaml
 from huggingface_hub import hf_hub_download
 import os
+import torch
 
 class TrafficSignDetector:
     def __init__(self, config_path):
@@ -17,7 +18,9 @@ class TrafficSignDetector:
             repo_id = 'VietCat/GTSRB-Model'
             file_path = model_path.replace('VietCat/GTSRB-Model/', '')
             local_model_path = hf_hub_download(repo_id=repo_id, filename=file_path)
-            self.model = YOLO(local_model_path)
+            # Allow safe loading of ultralytics model
+            with torch.serialization.safe_globals([torch.nn.Module]):
+                self.model = YOLO(local_model_path)
         else:
             self.model = YOLO(model_path)