Update app.py

app.py

@@ -8,7 +8,7 @@ from ultralytics import YOLO
 # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 # Load the YOLOv8 model
-model = YOLO('yolov8x-seg.pt')
+model = YOLO('YOLOv8s.pt')
 
 def process_video(input_video_path):
     cap = cv2.VideoCapture(input_video_path)
@@ -33,7 +33,7 @@ def process_video(input_video_path):
         # Convert resized frame to torch tensor and move it to GPU
         frame_tensor = torch.from_numpy(resized_frame).permute(2, 0, 1).unsqueeze(0).float() / 255.0
 
-        threshold = 0.1
+        threshold = 0.2
         frame_copy = frame.copy()
         results = model(frame_tensor)[0]
         for result in results.boxes.data.tolist():