Added constraint to process 4 pages at max

pdf_to_image.py

@@ -12,14 +12,14 @@ from pdf2image.exceptions import (
 #     1)Images stored in output folder
 #     2)It returns path to stored images 
 #     '''
-#     images_paths = convert_from_path(pdfPath, 200, outputPath, fmt="jpeg", poppler_path=poppler_path, paths_only=True)
+#     images_paths = convert_from_path(pdfPath, 200, outputPath, fmt="jpeg", poppler_path=poppler_path, paths_only=True, last_page=4) # Only processing 4 pages at max 
 #     return images_paths
 
 # def pdfToImg2(pdfPath):
 #     '''
 #     1)Returns a list of Pillow images
 #     '''
-#     images = convert_from_path(pdfPath, 200, fmt="jpeg", poppler_path=poppler_path)
+#     images = convert_from_path(pdfPath, 200, fmt="jpeg", poppler_path=poppler_path, last_page=4) # Only processing 4 pages at max 
 #     return images
 
 # Changed version of above code for deployment on huggingface spaces 
@@ -28,12 +28,12 @@ def pdfToImg(pdfPath, outputPath):
     1)Images stored in output folder
     2)It returns path to stored images 
     '''
-    images_paths = convert_from_path(pdfPath, 200, outputPath, fmt="jpeg", paths_only=True)
+    images_paths = convert_from_path(pdfPath, 200, outputPath, fmt="jpeg", paths_only=True, last_page=4) # Only processing 4 pages at max
     return images_paths
 
 def pdfToImg2(pdfPath):
     '''
     1)Returns a list of Pillow images
     '''
-    images = convert_from_path(pdfPath, 200, fmt="jpeg")
+    images = convert_from_path(pdfPath, 200, fmt="jpeg", last_page=4) # Only processing 4 pages at max
     return images