We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 66d0de1 commit a747636Copy full SHA for a747636
omni-reader/schemas/image_description.py
@@ -0,0 +1,13 @@
1
+"""This module contains the schema for the OCR results."""
2
+
3
+from typing import List, Optional
4
5
+from pydantic import BaseModel, Field
6
7
8
+class ImageDescription(BaseModel):
9
+ """Base model for OCR results."""
10
11
+ raw_text: str = Field(description="Extracted text from the image")
12
+ description: str = Field(description="Description of the image")
13
+ entities: Optional[List[str]] = Field(default=None, description="List of entities found in the image")
0 commit comments