updates

app.py

@@ -5,16 +5,6 @@ import numpy as np
 from PIL import Image
 import os
 
-# Fix for PyTorch 2.6 weights_only issue
-torch.serialization.add_safe_globals([
-    'ultralytics.nn.tasks.DetectionModel',
-    'ultralytics.nn.modules.block.C2f',
-    'ultralytics.nn.modules.block.Bottleneck',
-    'ultralytics.nn.modules.conv.Conv',
-    'ultralytics.nn.modules.head.Detect',
-    'ultralytics.nn.modules.block.SPPF'
-])
-
 # Load YOLOv8 Small model
 print("Loading YOLOv8 Small model...")
 model = YOLO('yolov8s.pt')

requirements.txt

@@ -1,6 +1,6 @@
 ultralytics==8.0.196
-torch>=1.9.0
-torchvision>=0.10.0
+torch==2.0.1
+torchvision==0.15.2
 pillow>=8.0.0
 gradio>=4.0.0
 opencv-python-headless>=4.5.0