Error Deploying Private Endpoint

[Sergioft]

I’m trying to create a private endpoint (with AWS provider) using the huggingface UI in order to deploy a model from repo arpanghoshal/EmoRoBERTa… Unfortunately, I get the following error:

~ ValueError: Could not load model /repository with any of the following classes: (<class ‘transformers.models.auto.modeling_auto.AutoModelForSequenceClassification’>, <class ‘transformers.models.roberta.modeling_roberta.RobertaForSequenceClassification’>)

Given I’m using the inference deployment UI, what can I do to resolve this?

[philschmid]

@Sergioft

arpanghoshal/EmoRoBERTa

Have you made sure that the selected framework is “tensorflow”?

[Sergioft]

Yes I have. The framework is set to Tensorflow.