[feat] Add probability input demo