Tengo un modelo de transformador simple (BERT) para la clasificación de texto binario y lo exporté a tensorflowjs y lo importé a una extensión de Chrome. El modelo y los pesos se cargaron correctamente, pero no puedo usar el modelo para predecir una oración. A continuación, se muestra el código para predecir el texto de entrada del script de contenido.

 const prediction = res.executeAsync({'input_ids':inputTxt},[-1,-1],[-1,-1])

A continuación se muestran los nodos de entrada y las entradas del modelo (lo copié de la consola de Chrome):

 inputNodes: Array(3) 0: "attention_mask" 1: "input_ids" 2: "token_type_ids" length: 3 [[Prototype]]: Array(0) inputs: Array(3) 0: dtype: "int32" name: "attention_mask" shape: (2) [-1, -1] [[Prototype]]: Object 1: dtype: "int32" name: "input_ids" shape: (2) [-1, -1] [[Prototype]]: Object 2: dtype: "int32" name: "token_type_ids" shape: (2) [-1, -1] [[Prototype]]: Object length: 3 [[Prototype]]: Array(0)

¿Alguien sabe cuál es mi modelo que espera que ingrese? Como puedo usar model.predict() en python antes de convertirlo a Chrome Extension.