removed call to generator in predict

Author: dakshvar22

rasa/utils/tensorflow/models.py

@@ -206,8 +206,9 @@ def predict(self, predict_data: RasaModelData) -> Dict[Text, tf.Tensor]:
             logger.debug("There is no tensorflow prediction graph.")
             self.build_for_predict(predict_data)
 
-        predict_dataset = predict_data.as_tf_dataset(batch_size=1)
-        batch_in = next(iter(predict_dataset))
+        # predict_dataset = predict_data.as_tf_dataset(batch_size=1)
+        batch_in = predict_data.prepare_batch(predict_data.data, 0, 1)
+        # batch_in = next(iter(predict_dataset))
 
         self._training = False  # needed for eager mode
         return self._predict_function(batch_in)