fixes 563

Change-Id: Ifb9743822c744845aac4deea9963a0c6e6389362

Author: MarkDaoust

docs/build_docs.py

@@ -22,13 +22,11 @@
 $> python build_docs.py
 """
 
-import os
 import pathlib
 import re
 import textwrap
 import typing
 
-
 from absl import app
 from absl import flags
 
@@ -48,6 +46,7 @@
 
 from tensorflow_docs.api_generator import generate_lib
 from tensorflow_docs.api_generator import public_api
+from tensorflow_docs.api_generator import parser
 
 import yaml
 
@@ -74,6 +73,33 @@
     "where to find the project code",
 )
 
+parser.ITEMS_TEMPLATE = textwrap.dedent(
+    """\
+  <tr>
+  <td>
+
+  {name}{anchor}
+
+  </td>
+  <td>
+
+  {description}
+
+  </td>
+  </tr>"""
+)
+
+parser.TEXT_TEMPLATE = textwrap.dedent(
+    """\
+  <tr class="alt">
+  <td colspan="2">
+
+  {text}
+
+  </td>
+  </tr>"""
+)
+
 
 def gen_api_docs():
     """Generates api docs for the generative-ai package."""