Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import fitz  # PyMuPDF
 from PIL import Image
 from concurrent.futures import ThreadPoolExecutor, as_completed
-
+import spaces
 # Load the trained model globally
 model_path = 'best.pt'  # Replace with the path to your trained .pt file
 model = YOLO(model_path)
@@ -68,6 +68,7 @@ def process_single_page(page, low_dpi, high_dpi, scale_factor):
         print(f"Error processing page: {e}")
         return []
 
+@spaces.GPU
 def process_pdf(pdf_file):
     # Open the PDF file
     doc = fitz.open(pdf_file)