login

app.py

@@ -6,7 +6,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"
 
@@ -145,6 +145,9 @@ class Detection:
 
         return individual_graphs['Nicks'], individual_graphs['Dents'], individual_graphs['Scratches'], individual_graphs['Pittings']
 
+
+  
+
     
 
     def generate_frequency_graph(self, annotations):
@@ -207,7 +210,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