From 34cd304732fe213e3870ff1f38e1970bbb592f1f Mon Sep 17 00:00:00 2001
From: Jun Siang Cheah <git@jscheah.me>
Date: Mon, 16 Sep 2024 17:40:42 +0100
Subject: [PATCH] cleanups

---
 README.md                    |  3 +--
 src/pull_available_models.py | 14 ++++++++++----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 586520d..deb8a0a 100644
--- a/README.md
+++ b/README.md
@@ -101,11 +101,10 @@ This lists various services that provide free access or credits towards API-base
             <td rowspan="2"><a href="https://cohere.com" target="_blank">Cohere</a></td>
             <td rowspan="2"><a href="https://docs.cohere.com/docs/rate-limits">20 requests/min<br>1000 requests/month</a></td>
             <td>Command-R</td>
-            <td>Shared Limit</td>
+            <td rowspan="2">Shared Limit</td>
         </tr>
         <tr>
             <td>Command-R+</td>
-            <td>Shared Limit</td>
         </tr><tr>
             <td><a href="https://huggingface.co/docs/api-inference/en/index" target="_blank">HuggingFace Serverless Inference</a></td>
             <td>Limited to models smaller than 10GB.<br>Some popular models are supported even if they exceed 10GB.</td>
diff --git a/src/pull_available_models.py b/src/pull_available_models.py
index f435c3b..495a0da 100644
--- a/src/pull_available_models.py
+++ b/src/pull_available_models.py
@@ -575,11 +575,10 @@ def main():
             <td rowspan="2"><a href="https://cohere.com" target="_blank">Cohere</a></td>
             <td rowspan="2"><a href="https://docs.cohere.com/docs/rate-limits">20 requests/min<br>1000 requests/month</a></td>
             <td>Command-R</td>
-            <td>Shared Limit</td>
+            <td rowspan="2">Shared Limit</td>
         </tr>
         <tr>
             <td>Command-R+</td>
-            <td>Shared Limit</td>
         </tr>"""
 
     table += """<tr>
@@ -646,8 +645,15 @@ def main():
 
     trial_table = ""
     for idx, model in enumerate(hyperbolic_models):
-        trial_table += f"<tr>{f'<td rowspan="{len(hyperbolic_models)}"><a href="https://app.hyperbolic.xyz/" target="_blank">Hyperbolic</a></td>' if idx == 0 else ''}{ f'<td rowspan="{len(hyperbolic_models)}">$10</td><td rowspan="{len(hyperbolic_models)}"></td>' if idx == 0 else ''}<td>{model['name']}</td></tr>\n"
-
+        trial_table += "<tr>"
+        if idx == 0:
+            trial_table += f'<td rowspan="{len(hyperbolic_models)}">'
+            trial_table += '<a href="https://app.hyperbolic.xyz/" target="_blank">Hyperbolic</a>'
+            trial_table += '</td>'
+            trial_table += f'<td rowspan="{len(hyperbolic_models)}">$10</td>'
+            trial_table += f'<td rowspan="{len(hyperbolic_models)}"></td>'
+        trial_table += f"<td>{model['name']}</td>"
+        trial_table += "</tr>\n"
     if MISSING_MODELS:
         logger.warning("Missing models:")
         logger.warning(list(MISSING_MODELS))