hi there, thanks for your work, I use your code in training my data, and it works well, but it seems doesn't have inference function, could you provide a copy or just give some insprit ? thanks!!