Update app.py

app.py

@@ -4,9 +4,11 @@ import gradio as gr
 import pandas as pd
 import torch
 import json
+from ultralytics import YOLO
 
 # Load the model
-detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
+# detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
+detect_model = YOLO('yolov8m_2023-10-23_best.pt')
 
 def detect(image):
   # Inference