now with label title

app.py

@@ -13,19 +13,19 @@ import pandas as pd
 BENCHMARK_DATA = {
     "Greedy Search": {
         "DistilGPT2": {
-            "T4": [],
-            "3090": [],
-            "A100": [],
+            "T4": [336.22, 3976.23, 115.84],
+            "3090": [158.38, 1835.82, 46.56],
+            "A100": [371.49, 4073.84, 60.94],
         },
         "GPT2": {
-            "T4": [],
-            "3090": [],
-            "A100": [],
+            "T4": [607.31, 7140.23, 185.12],
+            "3090": [297.03, 3308.31, 76.68],
+            "A100": [691.75, 7323.60, 110.72],
         },
         "OPT-1.3B": {
-            "T4": [],
-            "3090": [],
-            "A100": [],
+            "T4": [1303.41, 15939.07, 1488.15],
+            "3090": [428.33, 7259.43, 468.37],
+            "A100": [1125.00, 16713.63, 384.52],
         },
         "GPTJ-6B": {
             "T4": [],
@@ -152,7 +152,8 @@ def get_plot(model_name, generate_type):
     )
     g.despine(left=True)
     g.set_axis_labels("GPU", "Generation time (ms)")
-    g.legend.set_title(f"{generate_type} generation time for {model_name}")
+    g.set(title=f"{generate_type} generation time for {model_name} (no bar means OOM)")
+    g.legend.set_title("Framework")
     return plt.gcf()
 
 demo = gr.Blocks()
@@ -182,7 +183,7 @@ with demo:
                     ["Pytorch Version", "1.11.0"],
                     ["OS", "22.04 LTS (3090) / Debian 10 (other GPUs)"],
                     ["CUDA", "11.6 (3090) / 11.3 (others GPUs)"],
-                    ["Number of runs", "100 (the first run was discarded to ignore compilation time)"],
+                    ["Number of Runs", "100 (the first run was discarded to ignore compilation time)"],
                     ["Is there code to reproduce?", "Yes -- https://gist.github.com/gante/f0017e3f13ac11b0c02e4e4db351f52f"],
                 ],
             )