diff --git a/docling/models/picture_description_api_model.py b/docling/models/picture_description_api_model.py index a3c0c2ee..771608e2 100644 --- a/docling/models/picture_description_api_model.py +++ b/docling/models/picture_description_api_model.py @@ -51,7 +51,7 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): # Note: technically we could make a batch request here, # but not all APIs will allow for it. For example, vllm won't allow more than 1. def _api_request(image): - return api_image_request( + response, _ = api_image_request( image=image, prompt=self.options.prompt, url=self.options.url, @@ -60,5 +60,7 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): **self.options.params, ) + return response + with ThreadPoolExecutor(max_workers=self.concurrency) as executor: yield from executor.map(_api_request, images)