Welcome to the Julia discourse.
I am not familiar with machine learning libraries, but a quick check of the docs suggests that the loss function should take 2 arguments.
A general note: It is easier for others to help you if you follow some suggestions. In particular, a MWE with a complete error message would be useful.