Fix issue with llm service

marker/config/parser.py

@@ -86,7 +86,7 @@ def get_llm_service(self):
         if not self.cli_options.get("use_llm", False):
             return None
 
-        service_cls = self.cli_options["llm_service"]
+        service_cls = self.cli_options.get("llm_service", None)
         if service_cls is None:
             service_cls = "marker.services.gemini.GoogleGeminiService"
         return service_cls

marker/services/gemini.py

@@ -20,7 +20,7 @@ class BaseGeminiService(BaseService):
 
     def img_to_bytes(self, img: PIL.Image.Image):
         image_bytes = BytesIO()
-        img.save(image_bytes, format="PNG")
+        img.save(image_bytes, format="WEBP")
         return image_bytes.getvalue()
 
     def get_google_client(self, timeout: int):
@@ -45,7 +45,7 @@ def __call__(
             image = [image]
 
         client = self.get_google_client(timeout=timeout)
-        image_parts = [types.Part.from_bytes(data=self.img_to_bytes(img), mime_type="image/png") for img in image]
+        image_parts = [types.Part.from_bytes(data=self.img_to_bytes(img), mime_type="image/webp") for img in image]
 
         tries = 0
         while tries < max_retries:
@@ -57,7 +57,7 @@ def __call__(
                         "temperature": 0,
                         "response_schema": response_schema,
                         "response_mime_type": "application/json",
-                    }
+                    },
                 )
                 output = responses.candidates[0].content.parts[0].text
                 total_tokens = responses.usage_metadata.total_token_count

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "marker-pdf"
-version = "1.5.1"
+version = "1.5.2"
 description = "Convert PDF to markdown with high speed and accuracy."
 authors = ["Vik Paruchuri <github@vikas.sh>"]
 readme = "README.md"