change image model

ai-moderation/app/core/dependencies.py

@@ -2,6 +2,7 @@ from functools import lru_cache
 
 from app.ml.image.clip_classifier import ClipClassifier
 from app.ml.image.image_classifier import ImageClassifier
+from app.ml.image.weapon_detector import WeaponDetector
 from app.ml.text_classifier import TextClassifier
 from app.moderation.image.validator import ImageValidator
 from app.moderation.profanity.detector import ProfanityDetector
@@ -49,10 +50,16 @@ def get_image_classifier() -> ImageClassifier:
   return ImageClassifier()
 
 
+@lru_cache
+def get_weapon_detector() -> WeaponDetector:
+  return WeaponDetector()
+
+
 @lru_cache
 def get_image_moderation_service() -> ImageModerationService:
   return ImageModerationService(
       classifier=get_image_classifier(),
       validator=get_image_validator(),
-      clip_classifier=get_clip_classifier()
+      clip_classifier=get_clip_classifier(),
+      weapon_detector=get_weapon_detector()
   )