Update app.py

app.py

@@ -30,12 +30,6 @@ def crop_images_from_boxes(image, boxes, scale_factor):
     ]
     return cropped_images
 
-# Warm-up function to perform a dummy inference
-@spaces.GPU
-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
@@ -51,7 +45,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)
@@ -70,9 +64,6 @@ 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,