Update gradio.py

gradio.py

@@ -5,7 +5,7 @@ import numpy as np
 import torchvision.transforms as T
 
 # Load your model
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt')  # Adjust path if necessary
+model = torch.hub.load('ultralytics/yolov8', 'custom', path='best.pt')  # Adjust path if necessary
 model.eval()
 
 # Define your classes