SakibRumu commited on
Commit
8754806
·
verified ·
1 Parent(s): 6e3fa56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import cv2
4
  import pytesseract
5
 
6
  # Load the trained YOLOv5 model (replace 'best.pt' with your actual model path)
7
- model = torch.hub.load('ultralytics/yolov10n', 'custom', path='runs/train/exp/weights/yolov10n.pt')
8
 
9
  def process_video(input_video):
10
  # Read video frames
 
4
  import pytesseract
5
 
6
  # Load the trained YOLOv5 model (replace 'best.pt' with your actual model path)
7
+ model = torch.hub.load('ultralytics/yolov10n', 'custom', path='best.pt')
8
 
9
  def process_video(input_video):
10
  # Read video frames