AWS Deploy - Bert-Base-Uncased-Go-Emotion

🤗Transformers
1

I need some help with deploying Bert-Base-Uncased-Go-Emotion on AWS. How do I get the model to “return_all_scores” in AWS Sagemaker.

hub = {‘HF_MODEL_ID’:‘bhadresh-savani/bert-base-go-emotion’,
‘HF_TASK’:‘text-classification’}

thank you for your help