fix: remove unused httpx (#919)

* remove unused httpx Signed-off-by: Michele Dolfi <[email protected]> * use requests instead of httpx Signed-off-by: Michele Dolfi <[email protected]> * remove more usage of httpx Signed-off-by: Michele Dolfi <[email protected]> --------- Signed-off-by: Michele Dolfi <[email protected]>
DS4SD · Feb 7, 2025 · c18f47c · c18f47c
1 parent 4cc6e3e
commit c18f47c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py
@@ -4,9 +4,7 @@
 from pathlib import Path
 from typing import Iterable, List, Optional
 
-import httpx
 import numpy
-import torch
 from docling_core.types.doc import BoundingBox, CoordOrigin
 
 from docling.datamodel.base_models import Cell, OcrCell, Page

diff --git a/docling/models/picture_description_api_model.py b/docling/models/picture_description_api_model.py
@@ -3,11 +3,7 @@
 import logging
 from typing import Iterable, List, Optional
 
-import httpx
-from docling_core.types.doc import PictureItem
-from docling_core.types.doc.document import (  # TODO: move import to docling_core.types.doc
-    PictureDescriptionData,
-)
+import requests
 from PIL import Image
 from pydantic import BaseModel, ConfigDict
 
@@ -90,13 +86,13 @@ def _annotate_images(self, images: Iterable[Image.Image]) -> Iterable[str]:
                 **self.options.params,
             }
 
-            r = httpx.post(
+            r = requests.post(
                 str(self.options.url),
                 headers=self.options.headers,
                 json=payload,
                 timeout=self.options.timeout,
             )
-            if not r.is_success:
+            if not r.ok:
                 _log.error(f"Error calling the API. Reponse was {r.text}")
             r.raise_for_status()