Update app.py

app.py

@@ -1,20 +1,14 @@
 import gradio as gr
-from ultralytics import YOLO
 from PIL import Image, ImageDraw
 import io
 from fpdf import FPDF
 import tempfile
 import cv2
 import os
-import torch.serialization
+import torch
 import numpy as np
 from urllib.request import urlretrieve
-from ultralytics.nn.modules import Conv, Bottleneck, SPPF, C2f, Detect
-from ultralytics.nn.tasks import DetectionModel
-from torch.nn.modules.container import Sequential
-
-# Allowlist all necessary YOLOv8 modules to avoid UnpicklingError
-torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv, Bottleneck, SPPF, C2f, Detect])
+from ultralytics.yolo.engine.model import YOLO
 
 # Model path
 model_path = 'model/yolov8n.pt'
@@ -25,7 +19,7 @@ if not os.path.exists(model_path):
     url = 'https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt'
     urlretrieve(url, model_path)
 
-# Load the YOLOv8 model
+# Load the YOLOv8 model with weights_only=False internally handled by YOLO class
 model = YOLO(model_path)
 
 def detect_empty_spots(img_width, img_height, bottle_bboxes, min_gap=50):