Update app.py

app.py

@@ -5,7 +5,7 @@ import cv2
 import numpy as np
 import fitz  # PyMuPDF
 from PIL import Image
-
+import spaces
 # Load the trained model
 model_path = 'best.pt'  # Replace with the path to your trained .pt file
 model = YOLO(model_path)
@@ -43,6 +43,7 @@ def crop_images_from_boxes(image, boxes, scale_factor):
         cropped_images.append(cropped_image)
     return cropped_images
 
+@spaces.gpu
 def process_pdf(pdf_file):
     # Open the PDF file
     doc = fitz.open(pdf_file)