Update app.py

app.py

@@ -7,6 +7,7 @@ from ultralytics import YOLO
 import supervision as sv
 from PIL import Image
 from huggingface_hub import snapshot_download
+import spaces  # Import spaces for ZeroGPU compatibility
 
 # Adjustable parameters for detection
 CONFIDENCE_THRESHOLD = 0.1  # Confidence threshold for detections
@@ -27,6 +28,7 @@ model_path = os.path.join(model_dir, "best.pt")  # Adjust if filename differs
 model = YOLO(model_path).to(device)
 
 # Define the detection function for Gradio
+@spaces.GPU  # Decorator to allocate GPU for ZeroGPU-enabled Spaces
 def detect_objects(image: np.ndarray) -> Image.Image:
     # Ensure the image is in BGR format if provided by PIL (Gradio gives us an RGB image)
     if image.shape[-1] == 3:
@@ -82,7 +84,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_img = gr.Image(label="Detection Result", interactive=False)
     
-# Add Examples section with images from the root directory
+    # Add Examples section with images from the root directory
     with gr.Accordion("Select an Example Image"):
         example_root = os.path.dirname(__file__)  # Root directory
         example_images = [os.path.join(example_root, file) for file in os.listdir(example_root) if file.endswith(".jpg")]