added image ai-moderation

ai-moderation/app/test/test_text_classifier.py

@@ -1,20 +1,20 @@
-from app.ml.model_manager import model_manager
-from app.services.text_moderation_service import TextModerationService
+from app.container import profanity_detector
 
-model_manager.initialize()
 
-service = TextModerationService()
-
-texts = [
-  "Привет",
-  "Спасибо",
+tests = [
+  "ЗАлУпа",
   "Ты идиот",
+  "Это идиоты",
   "Хорошего дня"
 ]
 
-for text in texts:
-  result = service.moderate(text)
 
-  print("=" * 50)
-  print(text)
-  print(result)
+for text in tests:
+
+  result = profanity_detector.detect(text)
+
+  print(
+      text,
+      "=>",
+      result
+  )