Abby FlexiCapture Classifier not working in UiPath Studio

Hello, I created a classifier in FlexiLayout Studio and it is working fine there (successfully classifying docs). However, using the classifier in UiPath Studio is always resulting in “unknown” given that I am testing it on the same training & validation set used in FlexiLayout Studio. Anyone faced this issue before?