Update app.py

app.py

@@ -5,11 +5,11 @@ from PIL import Image
 
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True)
 
-def yolo(im):
+def detect(image):
     #g = (size / max(im.size))  # gain
    # im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
 
-    results = model(im)  # inference
+    results = model(image)  # inference
     results.render()  # updates results.imgs with boxes and labels
     return Image.fromarray(results.imgs[0])
 
@@ -20,5 +20,5 @@ outputs = gr.outputs.Image(type="pil", label="Output Image")
 title = "Object detection from Infrared image using YOLOv5n"
 
 
-gr.Interface(im, inputs, outputs, title=title, theme="huggingface").launch(debug=True)
+gr.Interface(detect, inputs, outputs, title=title, theme="huggingface").launch(debug=True)