Upload folder using huggingface_hub

app.py

@@ -147,10 +147,10 @@ def segment_image_core(img: np.ndarray | Image.Image) -> Image.Image:
     return (img, sections), jpg_img
 
 
-def pdf_to_image(pdf, end, start = 0) -> list[Image.Image]:
+def pdf_to_image(pdf, end = None, start = 0) -> list[Image.Image]:
     pdf = pdfium.PdfDocument(pdf)
     page_images = []
-    if not end:
+    if end is None:
         end = len(pdf)  # get the number of pages in the document
     for i in range(start, end):
         page = pdf[i]
@@ -202,7 +202,7 @@ with gr.Blocks() as demo:
 @app.post("/uploadPdf", response_class=FileResponse)
 def extract_image(uploadFile: UploadFile) -> FileResponse:
     file = uploadFile.file.read()
-    image = pdf_to_image(file)
+    image = pdf_first_page_to_image(file)
     img_segments, jpeg_image = segment_image_core(image)
     id = str(uuid.uuid4())
     filename = f"{cropped_dir}/cropped_{id}.jpg"