Update app.py

app.py

@@ -435,7 +435,8 @@ class IntrusionDetectionEn:
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
         self.model = YOLO(self.model_path)
         self.model.to(self.device)
-    
+        
+    @spaces.GPU
     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)
@@ -448,7 +449,6 @@ class IntrusionDetectionEn:
                 return True
         return False
     
-    @spaces.GPU
     def intrusion_detect_en(self, video_path):
         try:
             cap = cv2.VideoCapture(video_path)