Update model classifier

src/models/bert.py

@@ -25,7 +25,7 @@ class BERTClassifier():
         
         if model_name.startswith('jeevavijay10'):
             # self.model = torch.load(model_name)
-            self.model = AutoModel.from_pretrained(model_name)
+            self.model = BertForSequenceClassification.from_pretrained(model_name)
         else:
             self.model = BertForSequenceClassification.from_pretrained(
                 self.model_name,