diff --git a/docling/models/easyocr_model.py b/docling/models/easyocr_model.py index 9b1b2a0..0eccb98 100644 --- a/docling/models/easyocr_model.py +++ b/docling/models/easyocr_model.py @@ -4,9 +4,7 @@ import zipfile 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 index 6c7e02f..86b7694 100644 --- a/docling/models/picture_description_api_model.py +++ b/docling/models/picture_description_api_model.py @@ -3,11 +3,7 @@ import io 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 @@ class PictureDescriptionApiModel(PictureDescriptionBaseModel): **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()