I would like to use the DIET classifier in conjunction with cleanlab.ai. It is a tool to detect labeling errors and requires the model to be passed in as an argument. Please see here: examples/huggingface_keras_imdb.ipynb at master · cleanlab/examples · GitHub
Is it possible to do so using the RASA DIET?