Glad you got it working with the await! Indeed the function can be called on Tensors so as long as what is returned is a Tensor then it should be able to call data on that.  See TF Tensor API here:

  
      

      js.tensorflow.org
  

  
    

TensorFlow.js

  A WebGL accelerated, browser based JavaScript library for training and deploying ML models