Update app.py

app.py

@@ -1,9 +1,17 @@
 import gradio as gr
 import torch
-
 from PIL import Image
 
-model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True)
+model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', source="local")
+
+
+def yolo(im, size=640):
+    g = (size / max(im.size))  # gain
+    im = im.resize((int(x * g) for x in im.size), Image.BICUBIC)  # resize
+
+    results = model(im)  # inference
+    results.render()  # updates results.imgs with boxes and labels
+    return Image.fromarray(results.imgs[0])
 
 def detect(image):
     results = model(image)