SakibRumu commited on
Commit
80ae37c
·
verified ·
1 Parent(s): 94889fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -8,12 +8,11 @@ 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
 
8
  import os
9
 
10
  import sys
11
+ sys.path.append('/home/user/app/yolov10') # Add your local YOLOv10 repo path
12
 
13
+ from ultralytics import YOLO # This now refers to your local repo, not pip
14
 
15
+ model = YOLO('/home/user/app/best.pt') # Load the model
 
16
  model.eval()
17
 
18
  # Frame processing function