diff --git a/truss/cli/cli.py b/truss/cli/cli.py index 3c0628167..2bf5b9746 100644 --- a/truss/cli/cli.py +++ b/truss/cli/cli.py @@ -353,7 +353,7 @@ def predict( result = service.predict(request_data) if inspect.isgenerator(result): for chunk in result: - rich.print(chunk, end="") + click.echo(chunk, nl=False) return rich.print_json(data=result)