⬆️ [Update] dependency and the load cache function

requirements-dev.txt

@@ -1,4 +1,5 @@
 -r requirements.txt
+gradio
 pytest
 pytest-cov
 pre-commit

yolo/tools/data_loader.py

@@ -52,7 +52,7 @@ class YoloDataset(Dataset):
             data = self.filter_data(dataset_path, phase_name)
             torch.save(data, cache_path)
         else:
-            data = torch.load(cache_path)
+            data = torch.load(cache_path, weights_only=False)
             logger.info("📦 Loaded {} cache", phase_name)
         return data