As you already released the pre-trained model, could you please provide the source code?
@Exely I think the author shares a simple version, recently via Colab. check here.