Update app.py

app.py

@@ -5,6 +5,11 @@ import gradio as gr
 import uuid
 import os
 import torch
+from ultralytics import YOLO
+import torch.serialization
+
+with torch.serialization.safe_globals([YOLO]):
+    model = YOLO("best.pt")
 
 
 # Load your YOLOv8 model