Thanks @lgusm for your kind words and suggestion.
Due to the fact that this was an example I did not train the model with much data to give me high quality results (as one might notice in the hugging face demo).
I will definitely contribute to the hub once I have a better model to share. Thanks again