schedule function custom_item_model_data:second 1s

execute as @a[scores={CustomModel=1}] if items entity @s weapon.mainhand * run function custom_item_model_data:trigger
execute as @a[scores={CustomModel=2..}] if items entity @s weapon.mainhand * run function custom_item_model_data:remove_model

# resets trigger
execute as @a run function custom_item_model_data:enable