Add Docker build and deploy pipeline
All checks were successful
deploy-ai / deploy (push) Successful in 2m56s
All checks were successful
deploy-ai / deploy (push) Successful in 2m56s
This commit is contained in:
14
ai-moderation/app/ml/image/weapon_prediction_result.py
Normal file
14
ai-moderation/app/ml/image/weapon_prediction_result.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
class WeaponPredictionResult:
|
||||
label: str
|
||||
|
||||
score: float
|
||||
|
||||
detected_labels: list[str] = field(
|
||||
default_factory=list
|
||||
)
|
||||
|
||||
box_count: int = 0
|
||||
Reference in New Issue
Block a user