Update app.py

app.py

@@ -4,7 +4,7 @@ matplotlib.use('Agg')
 import streamlit as st
 import cv2
 import numpy as np
-from yolov5 import YOLOv5
+#from yolov5 import YOLOv5
 from sort.sort import Sort
 import tempfile
 import shutil
@@ -12,7 +12,7 @@ from moviepy.editor import VideoFileClip, concatenate_videoclips, ImageSequenceC
 import os
 
 # Load the pre-trained model and initialize the SORT tracker
-model_path = 'yolov5s.pt'  # Ensure this path points to the model file
+model_path = 'best.pt'  # Ensure this path points to the model file
 model = YOLOv5(model_path, device='cpu')
 tracker = Sort()