Update README.md

README.md

@@ -97,6 +97,8 @@ pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
 
 ```python
 from ultralyticsplus import YOLO, render_result
+import torch
+
 
 # load model
 model = YOLO('foduucom/Tyre-Quality-Classification-AI')
@@ -108,13 +110,14 @@ model.overrides['agnostic_nms'] = False  # NMS class-agnostic
 model.overrides['max_det'] = 1000  # maximum number of detections per image
 
 # set image
-image = '/path/to/your/document/image'
+image = '/content/90-100-10-ceat-500x500.jpeg'
 
 # perform inference
-results = model.predict(image)
+results = model(image)
+
 
 # observe results
-top_class_index = results[0].probs.top1
+top_class_index = torch.argmax(results[0].probs).item()
 Class = model.names[top_class_index]
 print(Class)