Hugging Face's logo

Spaces:
ModuMLTECH
/
projet_trafic_2
Sleeping

App Files Community
ModuMLTECH commited on
Commit
d020b2d
·
verified ·
1 Parent(s): c81bd66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import os
13
 
14
  # Load the pre-trained model and initialize the SORT tracker
15
  model_path = 'best.pt' # Ensure this path points to the model file
16
- model = YOLOv5(model_path, device='cpu')
17
  tracker = Sort()
18
 
19
  def process_video(uploaded_file):
 
13
 
14
  # Load the pre-trained model and initialize the SORT tracker
15
  model_path = 'best.pt' # Ensure this path points to the model file
16
+ model = YOLO(model_path, device='cpu')
17
  tracker = Sort()
18
 
19
  def process_video(uploaded_file):