Update app.py

app.py

@@ -415,6 +415,13 @@ import os
 from ultralytics import YOLO
 import gradio as gr
 
+import torch
+import cv2
+import numpy as np
+import os
+from ultralytics import YOLO
+import gradio as gr
+
 class IntrusionDetectionEn:
     def __init__(self, model_path="yolov8n.pt", max_intrusion_time=300, iou_threshold=0.5, conf_threshold=0.7):
         self.model_path = model_path
@@ -435,8 +442,7 @@ 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)
@@ -449,6 +455,7 @@ class IntrusionDetectionEn:
                 return True
         return False
     
+    @gr.GPU
     def intrusion_detect_en(self, video_path):
         try:
             cap = cv2.VideoCapture(video_path)
@@ -505,6 +512,7 @@ class IntrusionDetectionEn:
 
 
 
+
 import cv2
 import numpy as np
 from ultralytics import YOLO