diff --git a/chandra/model/__init__.py b/chandra/model/__init__.py index 0b4b9e8..b703a18 100644 --- a/chandra/model/__init__.py +++ b/chandra/model/__init__.py @@ -48,6 +48,7 @@ class InferenceManager: page_box=[0, 0, input_item.image.width, input_item.image.height], token_count=result.token_count, images=extract_images(result.raw, chunks, input_item.image), + error=result.error, ) ) return output diff --git a/chandra/model/schema.py b/chandra/model/schema.py index 623a958..b6b75fa 100644 --- a/chandra/model/schema.py +++ b/chandra/model/schema.py @@ -27,3 +27,4 @@ class BatchOutputItem: page_box: List[int] token_count: int images: dict + error: bool