Update app.py

app.py

@@ -30,6 +30,11 @@ def crop_images_from_boxes(image, boxes, scale_factor):
     ]
     return cropped_images
 
+# Warm-up function to perform a dummy inference
+def warm_up_model():
+    dummy_image = np.zeros((640, 512, 3), dtype=np.uint8)
+    infer_image_and_get_boxes(dummy_image)
+
 @spaces.GPU
 def process_pdf(pdf_file):
     # Open the PDF file
@@ -45,7 +50,7 @@ def process_pdf(pdf_file):
 
     # Pre-cache all page pixmaps at low DPI
     low_res_pixmaps = [page.get_pixmap(dpi=low_dpi) for page in doc]
-    
+
     # Loop through each page
     for page_num, low_res_pix in enumerate(low_res_pixmaps):
         low_res_img = np.frombuffer(low_res_pix.samples, dtype=np.uint8).reshape(low_res_pix.height, low_res_pix.width, 3)
@@ -64,6 +69,9 @@ def process_pdf(pdf_file):
     
     return all_cropped_images
 
+# Warm-up the model before launching the app
+warm_up_model()
+
 # Create Gradio interface
 iface = gr.Interface(
     fn=process_pdf,