Dede16 commited on
Commit
28c8e1e
·
verified ·
1 Parent(s): d81f459

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from huggingface_hub import (
11
  )
12
 
13
  # Initialize the YOLO model
14
- model = YOLO('runs/segment/train/weights/best.pt')
15
  import os
16
 
17
  hf_token = os.environ.get('HUGGINGFACE_TOKEN')
 
11
  )
12
 
13
  # Initialize the YOLO model
14
+ model = YOLO('best.pt')
15
  import os
16
 
17
  hf_token = os.environ.get('HUGGINGFACE_TOKEN')