Update app.py

app.py

@@ -1,9 +1,9 @@
-import torch
 import gradio as gr
 from PIL import Image
 import numpy as np
+from ultralytics import YOLO
 
-model = torch.hub.load('.', 'custom', path=r'./best.pt', force_reload=True, source='local')
+model = YOLO('./best.pt')
 
 def yolo(im, size=512):
     g = (size / max(im.size))  # gain