А ночью я все же добил эту нейросеть. Надо теперь сделать интерфейс, но не хватает времени... посмотрим...