Update app.py

app.py

@@ -26,6 +26,7 @@ def infer_image_and_get_boxes(image, confidence_threshold=0.6):
     return boxes
 
 # Function to crop images from the boxes
+@spaces.GPU
 def crop_images_from_boxes(image, boxes, scale_factor):
     cropped_images = [
         image[int(y1 * scale_factor):int(y2 * scale_factor), int(x1 * scale_factor):int(x2 * scale_factor)]
@@ -33,7 +34,7 @@ def crop_images_from_boxes(image, boxes, scale_factor):
     ]
     return cropped_images
 
-
+@spaces.GPU
 def process_pdf(pdf_file):
     # Open the PDF file
     doc = fitz.open(pdf_file)