ai-accelerator-1.py

from sagemaker.tensorflow.model import TensorFlowModel, TensorFlowPredictor

prefix = 'keras_models'
s3_model_path = sess.upload_data(path='model.tar.gz', key_prefix=prefix)

model = TensorFlowModel(model_data=s3_model_path, 
                        framework_version='1.15',
                        role=role,
                        predictor_cls = TensorFlowPredictor,
                        sagemaker_session=sess)
 
predictor_gpu = model.deploy(initial_instance_count=1, 
                         instance_type='ml.g4dn.xlarge')