from image_prediction.image_extractor.extractors.parsable import ParsablePDFImageExtractor def extract_images_from_pdf(pdf, extractor=None): if not extractor: extractor = ParsablePDFImageExtractor() try: images_extracted, metadata_extracted = zip(*extractor(pdf)) return images_extracted, metadata_extracted except ValueError: return [], []