Update app.py

app.py

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