changes

app.py

@@ -6,6 +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"
 
@@ -208,6 +209,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

model/company_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e2510291d55581f170275335dccbab9c2b91d85db4602bc399e15a0f7a24662
+size 5461843