Changing vocabulary

[benakhovan32561]

I want to use other classes for my specific task.
I changed the class list in "th139_st101.py" and used gen_cls.py to generate the embeddings
But when I run the code afterwards it fails.
I can expand if needed but I wonder if I just did something wrong.

Making it easier to use custom classes will be nice

[HarborYuan]

Hi @benakhovan32561 ,

Can you provide the details about your error? I have verified that this script works fine in my own environment.