Hey @dtransposed Thanks for the code. Could you please share the pretrained model for this code? I don't have the resources to train it myself.