Spaces:

SakibHasan
/

license_plate_classifier2

Running

SakibRumu commited on Apr 6

Commit

a1ed15a

verified ·

1 Parent(s): dee6c1a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import cv2
 import pytesseract
 # Load the trained YOLOv5 model (replace 'best.pt' with your actual model path)
-model = torch.hub.load('ultralytics/yolov10n', 'custom', path='best(3).pt')
 def process_video(input_video):
     # Read video frames

 import pytesseract
 # Load the trained YOLOv5 model (replace 'best.pt' with your actual model path)
+model = torch.hub.load('ultralytics/yolov10n', 'custom', path='best(3).pt',map_location= 'cpu')
 def process_video(input_video):
     # Read video frames