Репозиторий сверточной нейросети для модерации на основе ruBERT. Код для данной нейросети был взят с ai_server. Изначально этот код должен был использоваться для классификации эмоций персонажей, но из-за большого количества параметров и большого объема модели не удалось реализовать real-time классификацию нескольких диалогов одновременно. Скорее всего, модель нужно сделать однослойной, этого будет вполне достаточно для классификации диалогов.
$ poetry install
$ poetry run uvicorn topic_moderator.main:app --host 0.0.0.0 --port 8000