Upload app.py

app.py

@@ -3,11 +3,12 @@ from ultralytics import YOLO
 import numpy as np
 import cv2
 
-@spaces.GPU
+
 # Load models
 model = YOLO("best-3.pt")  # load a custom model for segmentation (protection zone)
 model2 = YOLO('yolo11s.pt')  # load a second model for object detection
 
+@spaces.GPU
 def process_image(image):
     # Gradio passes the image as RGB, so we'll work in RGB color space
     image_rgb = np.array(image)