models/full_finetune_model.py

from transformers import BertForSequenceClassification

def get_full_finetune_model():
    model = BertForSequenceClassification.from_pretrained('bert-base-uncased')
    return model