docs: def translate_text(estimator, subtokenizer, txt):
    """Translate a single string."""
    encoded_txt = _encode_and_add_eos(txt, subtokenizer)
  
    def input_fn():
      ds = tf.data.Dataset.from_tensors(encoded_txt)
      ds = ds.batch(_DECODE_BATCH_SIZE)
      return ds
  
    predictions = estimator.predict(input_fn)
    translation = next(predictions)["outputs"]
    translation = _trim_and_decode(translation, subtokenizer)
    print("Translation of \"%s\": \"%s\"" % (txt, translation))