Files
post-moderation/ai-moderation/app/test/test_text_classifier.py
2026-07-17 01:06:36 +03:00

20 lines
383 B
Python

from app.ml.model_manager import model_manager
from app.services.text_moderation_service import TextModerationService
model_manager.initialize()
service = TextModerationService()
texts = [
"Привет",
"Спасибо",
"Ты идиот",
"Хорошего дня"
]
for text in texts:
result = service.moderate(text)
print("=" * 50)
print(text)
print(result)