Update news_category_prediction.py

news_category_prediction.py

@@ -34,7 +34,7 @@ def inference(text, interpreter, label_encoder, tokenizer):
         interpreter.set_tensor(input_details[1]["index"], input_ids)
         interpreter.invoke()
         tflite_pred = interpreter.get_tensor(output_details["index"])
-        tflite_pred = parse_prediction(tflite_pred)
+        tflite_pred = parse_prediction(tflite_pred, label_encoder)
     logger.warning('Exiting inference()')
     return tflite_pred