Update README.md

(cherry picked from commit e5d7122)
Lechtr · Apr 25, 2021 · c5c373b · c5c373b
1 parent 504d5a5
commit c5c373b
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -121,19 +121,18 @@ Done. (0.103s)
 
 ### PyTorch Hub
 
-To run **batched inference** with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36):
+Inference with YOLOv5 and [PyTorch Hub](https://github.com/ultralytics/yolov5/issues/36):
 ```python
 import torch
 
 # Model
 model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
 
-# Images
-dir = 'https://github.com/ultralytics/yolov5/raw/master/data/images/'
-imgs = [dir + f for f in ('zidane.jpg', 'bus.jpg')]  # batch of images
+# Image
+img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
 
 # Inference
-results = model(imgs)
+results = model(img)
 results.print()  # or .show(), .save()
 ```