SakibRumu commited on
Commit
c2f9c01
·
verified ·
1 Parent(s): 89b7cc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -7,12 +7,13 @@ from PIL import Image
7
  import sys
8
  import os
9
 
10
- # Add YOLOv10 repo to path
11
- sys.path.append('/home/user/app/yolov10') # adjust path if needed
12
- from models.common import DetectMultiBackend # YOLOv10 model loader
 
13
 
14
  # Load model
15
- model = DetectMultiBackend('/home/user/app/best.pt', device='cpu') # your trained YOLOv10n model path
16
  model.eval()
17
 
18
  # Frame processing function
 
7
  import sys
8
  import os
9
 
10
+ import sys
11
+ sys.path.append('/home/user/app/yolov10') # Add the root of your custom YOLOv10 repo to the path
12
+
13
+ from ultralytics.models.common import DetectMultiBackend # Import from your custom repo
14
 
15
  # Load model
16
+ model = DetectMultiBackend('/home/user/app/best.pt', device='cpu') # Update model path if needed
17
  model.eval()
18
 
19
  # Frame processing function