login

app.py

@@ -506,7 +506,7 @@ from sahi import AutoDetectionModel
 from PIL import Image
 import plotly.graph_objects as go
 import torch
-
+import spaces
 
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
@@ -709,7 +709,7 @@ def upload_image(image):
     """Process the uploaded image (if needed) and display it.""" 
     return image
 
-
+@spaces.GPU
 def apply_detection(image):
     """Run object detection on the uploaded image and return the annotated image.""" 
     # Convert image from PIL to NumPy array