06c50f050dc143126636b6c9b7d80ee3468377a6,examples/serving/ende_client.py,,main,#,82

Before Change

  stub = prediction_service_pb2_grpc.PredictionServiceStub(channel)
  tokenizer = pyonmttok.Tokenizer("none", sp_model_path=args.sentencepiece_model)

  batch_input = ["Hello world!", "My name is John.", "I live on the West coast."]
  batch_output = translate(stub, args.model_name, batch_input, tokenizer, timeout=args.timeout)
  for input_text, output_text in zip(batch_input, batch_output):
    print("{} ||| {}".format(input_text, output_text))


if __name__ == "__main__":
  main()

After Change

  tokenizer = pyonmttok.Tokenizer("none", sp_model_path=args.sentencepiece_model)

  while True:
    text = input("Source: ")
    output = translate(stub, args.model_name, [text], tokenizer, timeout=args.timeout)
    print("Target: %s" % output[0])
    print("")

Instances