Spaces:

akashrajeev
/

decodx

Sleeping

App Files Files Community

decodx / models.py

akashrajeev

Cursor

Deploy DecodX to Hugging Face Docker Space.

8f1a46c 6 months ago

Raw History Blame Contribute Delete

462 Bytes

from pydantic import BaseModel
from typing import List, Optional

class Req(BaseModel):
    documents: str
    questions: List[str]
    ocr_method: Optional[str] = "auto"  # NEW: OCR method selection

class ImageReq(BaseModel):
    """NEW: Request model for direct image upload"""
    questions: List[str]
    ocr_method: Optional[str] = "auto"

class Resp(BaseModel):
    answers: List[str]
    processing_info: Optional[dict] = None  # NEW: Processing metadata

	from pydantic import BaseModel
	from typing import List, Optional

	class Req(BaseModel):
	documents: str
	questions: List[str]
	ocr_method: Optional[str] = "auto" # NEW: OCR method selection

	class ImageReq(BaseModel):
	"""NEW: Request model for direct image upload"""
	questions: List[str]
	ocr_method: Optional[str] = "auto"

	class Resp(BaseModel):
	answers: List[str]
	processing_info: Optional[dict] = None # NEW: Processing metadata