better formatting output (#37441)

Author: ahmedabu98

examples/notebooks/beam-ml/run_inference_vllm.ipynb

@@ -231,7 +231,7 @@
       "source": [
         "class FormatOutput(beam.DoFn):\n",
         "  def process(self, element, *args, **kwargs):\n",
-        "    yield \"Input: {input}, Output: {output}\".format(input=element.example, output=element.inference)\n",
+        "    yield \"Input: {input}, Output: {output}\".format(input=element.example, output=element.inference.choices[0].text)\n",
         "\n",
         "prompts = [\n",
         "    \"Hello, my name is\",\n",
@@ -500,7 +500,7 @@
         "\n",
         "class FormatOutput(beam.DoFn):\n",
         "  def process(self, element, *args, **kwargs):\n",
-        "    yield \"Input: {input}, Output: {output}\".format(input=element.example, output=element.inference)\n",
+        "    yield \"Input: {input}, Output: {output}\".format(input=element.example, output=element.inference.choices[0].text)\n",
         "\n",
         "logging.getLogger().setLevel(logging.INFO)  # Output additional Dataflow Job metadata and launch logs. \n",
         "prompts = [\n",