Thanks for the amazing work! I am trying to build a prof of concept using huggingface models that might be replaced with custom build once later on. eleldar/theme-classification supports the arguments multi_class and candidate_labels but I am currently running into problems applying them correctly.

topics JSONB GENERATED ALWAYS AS (pgml.transform(
    	task   => '{
	       "task": "text-classification", 
               "model": "eleldar/theme-classification"
        }'::JSONB,
        args => '{ 
              "multi_class": true, 
              "candidate_labels": "[a, b,c]"
        }'::JSONB,
    	inputs => ARRAY[content]
)) STORED

This results in:

unexpected keyword argument 'multi_class'

What am i missing?