Update app.py

app.py

@@ -8,6 +8,7 @@ model_path = 'best.pt'  # Replace with the path to your trained .pt file
 model = YOLO(model_path)
 
 # Function to perform inference on an image
+@spaces.GPU
 def infer_image(image):
     # Convert the image from BGR to RGB
     image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)