Update app.py

app.py

@@ -2,9 +2,11 @@ import gradio as gr
 import torch
 from PIL import Image
 import requests
+from ultralytics import YOLO
+
 
 # Load the trained YOLOv8 model
-model = torch.hub.load('ultralytics/yolov8', 'custom', path='my_yolov8_model/best.pt')
+model = YOLO("./my_yolov8_model/best.pt")
 
 def detect_objects(image):
     # Run the YOLOv8 model on the input image