Based on and adapted from https://huggingface.co/spaces/nz03/nz03