aizenSosuke/SQL-PDFclassificationModel

This is classification model trained on RoBERTa base language model to classify a given text input into sql or docs. For a given question , if it classifies '0' it is of docs type and if it classifies as '1' , it is of sql type