Update app.py

app.py

@@ -11,7 +11,7 @@ def yolo(im, size=640):
     im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)  # resize
     results = model(im)  # inference
     results.render()  # updates results.imgs with boxes and labels
-    return Image.fromarray(results.imgs[0])
+    return Image.fromarray(results.ims[0])
 inputs = gr.inputs.Image(type='pil', label="Original Image")
 outputs = gr.outputs.Image(type="pil", label="Output Image")
 title = "Custom YOLOv5"