predict.js

async function predict() {
    const predictedClass = tf.tidy(() => {
      const text = document.getElementById("myText").value;
      const tokenisation = tokenise(text, word2index);
      const predictions = model.predict(tf.tensor2d(tokenisation, [1, maxLen]));
      return predictions.as1D().argMax();
    });
const classId = (await predictedClass.data())[0];
    var predictionText = "";
    switch(classId){
  case 0:
   predictionText = "Text is classified as Business News";
   break;
  case 1:
   predictionText = "Text is classified as Science-Technology News";
   break;
  case 2:
   predictionText = "Text is classified as Sport News";
   break;
  case 3:
   predictionText = "Text is classified as World News";
   break;
 }
 document.getElementById("prediction").innerText = predictionText;
predictedClass.dispose();
}