Where can I download the pre-trained model weights (checkpoint.pth)?

Hello, thanks for sharing this great work! I am trying to reproduce the project and run the Gradio demo, but I couldn't find the download link for the pre-trained checkpoint.pth mentioned in the README. Could you please provide a Hugging Face or Google Drive / Baidu Netdisk link for the model weights? Thanks!