Update app.py

app.py

@@ -414,7 +414,6 @@ class IntrusionDetectionEn:
         self.max_intrusion_time = max_intrusion_time
         self.iou_threshold = iou_threshold
         self.conf_threshold = conf_threshold
-    
 
         # Predefined staff uniform colors (RGB format)
         self.staff_colors = [
@@ -424,7 +423,11 @@ class IntrusionDetectionEn:
             (143, 147, 136),  # Gray-green
             (48, 59, 71)      # Dark blue/gray
         ]
-        
+
+        # 🔹 Load the model once
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.model = YOLO(self.model_path).to(self.device)
+
     def is_staff(self, person_crop):
         """Checks if the detected person is a staff member based on clothing color."""
         avg_color = np.mean(person_crop, axis=(0, 1))  # Compute average color (BGR)
@@ -437,18 +440,9 @@ class IntrusionDetectionEn:
                 return True
         return False
 
-        
     @spaces.GPU
     def intrusion_detect_en(self, video_path):
         try:
-            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-            if not os.path.exists(self.model_path):
-                model = YOLO("yolov8n.pt")
-                model.save(self.model_path)
-            else:
-                model = YOLO(self.model_path)
-            model.to(device)
-
             cap = cv2.VideoCapture(video_path)
             if not cap.isOpened():
                 raise ValueError(f"❌ Failed to open video: {video_path}")
@@ -472,7 +466,7 @@ class IntrusionDetectionEn:
                     break
                 frame_count += 1
 
-                results = model(frame)
+                results = self.model(frame)  # 🔹 Use preloaded model
                 for result in results:
                     boxes = result.boxes.xyxy.cpu().numpy()
                     classes = result.boxes.cls.cpu().numpy()
@@ -501,6 +495,7 @@ class IntrusionDetectionEn:
         except Exception as e:
             raise ValueError(f"Error in detect_intrusion: {str(e)}")
 
+
 import cv2
 import numpy as np
 from ultralytics import YOLO