32 lines
800 B
Python
32 lines
800 B
Python
from functools import lru_cache
|
|
|
|
from app.ml.text_classifier import TextClassifier
|
|
from app.services.text_moderation_service import TextModerationService
|
|
from app.ml.image.image_classifier import ImageClassifier
|
|
from app.services.image_moderation_service import ImageModerationService
|
|
from app.container import profanity_detector
|
|
|
|
@lru_cache
|
|
def get_text_classifier() -> TextClassifier:
|
|
return TextClassifier()
|
|
|
|
@lru_cache
|
|
def get_text_moderation_service() -> TextModerationService:
|
|
return TextModerationService(
|
|
classifier=get_text_classifier(),
|
|
profanity_detector=profanity_detector
|
|
)
|
|
|
|
@lru_cache
|
|
def get_image_classifier():
|
|
|
|
return ImageClassifier()
|
|
|
|
|
|
|
|
@lru_cache
|
|
def get_image_moderation_service():
|
|
|
|
return ImageModerationService(
|
|
classifier=get_image_classifier()
|
|
) |