Update tt.py

Author: Programmer-RD-AI

ML/modelling/tt.py

@@ -7,7 +7,7 @@ def __init__(
         num_classes: int = 2,
         input_dim: int = 768,
         classifier_head: torchtext.models = RobertaClassificationHead,
-        model: torchtext.models = XLMR_LARGE_ENCODER,
+        model: torchtext.models = ROBERTA_BASE_ENCODER,
     ) -> None:
         super().__init__()
         self.num_classes = num_classes