models.py 372 B

1234567891011121314
  1. from dataclasses import dataclass, field
  2. from datetime import datetime
  3. from typing import List, Optional
  4. @dataclass
  5. class PalmOilBunch:
  6. ripeness_class: str
  7. confidence: float
  8. embedding: List[float]
  9. box: List[float]
  10. image_data: str
  11. is_abnormal: bool = False
  12. timestamp: datetime = field(default_factory=datetime.now)
  13. id: Optional[str] = None