add latency filter

app.py

@@ -126,10 +126,13 @@ def get_benchmark_table(bench_df):
 
 
 def get_benchmark_plot(bench_df):
+    # filter latency bigger than 150s
+    bench_df = bench_df[bench_df["generate.latency(s)"] <= 150]
+
     fig = px.scatter(
         bench_df,
         y="best_score",
-        x="generate.throughput(tokens/s)",
+        x="generate.latency(s)",
         size="generate.peak_memory(MB)",
         color="model_type",
         custom_data=list(ALL_COLUMNS_MAPPING.keys()),
@@ -137,7 +140,7 @@ def get_benchmark_plot(bench_df):
     )
     fig.update_layout(
         title={
-            "text": "Model Score vs. Latency vs. Memory",
+            "text": "Latency vs. Score vs. Memory",
             "y": 0.95,
             "x": 0.5,
             "xanchor": "center",
@@ -145,7 +148,7 @@ def get_benchmark_plot(bench_df):
         },
         xaxis_title="Generation Throughput (tokens/s)",
         yaxis_title="Open LLM Score (%)",
-        legend_title="Model Type",
+        legend_title="LLM Type",
         width=1200,
         height=600,
     )