Update README.md

README.md

@@ -20,7 +20,7 @@ Below are the steps and code implementation.
 import cv2
 import matplotlib.pyplot as plt
 import numpy as np
-from ultralytics import YOLOv10
+from ultralytics import YOLO
 from google.colab.patches import cv2_imshow
 ```
 
@@ -34,7 +34,7 @@ from google.colab.patches import cv2_imshow
 
 ## Step 2: Load YOLOv10 Model
 ```python
-model = YOLOv10('vprashant/doclayout_detector/weights')  
+model = YOLO('vprashant/doclayout_detector/weights')  
 ```
 
 - Load the YOLOv10 model with the path to your trained weights.