Muito legal. Estou num projeto desses também, mas utilizando modelo do tipo sentence transformers.