def create_predictor()

in containers/serving/resources/predict.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def create_predictor(cls):
        try:
            model_dir = os.environ["SM_MODEL_DIR"]
        except KeyError:
            model_dir = "/opt/ml/model"
        
        cls.tokenizer = DistilBertTokenizerFast.from_pretrained(model_dir)
        cls.tf_model = TFDistilBertForSequenceClassification.from_pretrained(pretrained_model_name_or_path=model_dir)
        
        return cls.tf_model