Allow new users to select role

Hi Alex, I understand the approach, but that will be too troublesome for the users.

Possible if I use this? I have seen this somewhere, but I do not know how to customise it with mine.

//Create the user
$user_id = wp_create_user( $username, $password, $email );
$user = new WP_User( $user_id );

//If the user selects vendor set the role to vendor
if( $_POST[‘user_role’] == ‘vendor’ )
{
$user->set_role( ‘vendor’ );
}
//Else they are a customer
else
{
$user->set_role( ‘customer’ );
}

Could you help with this?