- Updated the sample code (f8754d06e54f1806a4eb097213eaa3bf3a668e40)


Co-authored-by: Frank Lee <frankleeeee@users.noreply.huggingface.co>