size

app.py

@@ -6,7 +6,7 @@ from sahi import AutoDetectionModel
 from PIL import Image
 import plotly.graph_objects as go
 import torch
-import spaces
+#import spaces
 
 import os
 import shutil
@@ -105,7 +105,7 @@ class Detection:
         # Populate the category_areas dictionary
         for annotation in annotations:
             category_name = annotation['category_name']
-            area = annotation['bbox'][2] * annotation['bbox'][3]  # Width * Height
+            area = annotation['area']
             if category_name in category_areas:
                 category_areas[category_name].append(area)
 
@@ -217,7 +217,7 @@ def upload_image(image):
     """Process the uploaded image (if needed) and display it.""" 
     return image
 
-@spaces.GPU
+#@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