# create Hugging Face Model Class
huggingface_model = HuggingFaceModel(
model_data= model_uri, # path to your trained sagemaker model
role=role, # iam role with permissions to create an Endpoint
transformers_version="4.17", # transformers version used
pytorch_version="1.10", # pytorch version used
py_version="py38",
env={ 'HF_TASK':'text-classification' }# python version of the DLC
)