fixed visualization bug

app.py

@@ -63,6 +63,10 @@ def create_prefill_visualization(filtered_data):
         hover_data=['Model', 'Dataset'],
         title='Prefill Stage: Quality vs TTFT (s) by Method'
     )
+    fig.update_layout(
+        yaxis=dict(range=[0, 100]),  # Set y-axis (Quality) range from 0 to 1
+        xaxis=dict(range=[0, None])  # Set x-axis (TTFT (s)) to start from 0
+    )
     return fig
 
 def create_decode_visualization(filtered_data):
@@ -76,6 +80,10 @@ def create_decode_visualization(filtered_data):
         hover_data=['Model', 'Dataset'],
         title='Decode Stage: Quality vs Throughput by Method'
     )
+    fig.update_layout(
+        yaxis=dict(range=[0, 100]),  # Set y-axis (Quality) range from 0 to 1
+        xaxis=dict(range=[0, None])  # Set x-axis (Throughput (token/s)) to start from 0
+    )
     return fig
 
 # Load the data from the /data folder
@@ -84,7 +92,6 @@ data = load_data(data_dir)
 
 # Gradio app UI and functionality
 def create_gradio_app():
-    print(plotly.__version__)
 
     with gr.Blocks() as app:
         with gr.Row():
@@ -197,8 +204,6 @@ This demo leaderboard allows users to explore and compare different KV cache imp
                 with gr.Row():
                     gr.Markdown("### Decode-Stage Visualization")
                 with gr.Row():
-                    print(create_decode_visualization(decode_default))
-
                     decode_plot = gr.Plot(
                         value=create_decode_visualization(decode_default)
                     )

data/decode_H2O_Mistral-7B-v0.3_LongGenBench.json

@@ -1,5 +1,5 @@
 {
-    "Quality": 0.90,
+    "Quality": 90,
     "Throughput (token/s)": 626.708,
     "Link": "http://example.com/decode"
 }

data/decode_StreamingLLM_Mistral-7B-v0.3_LongGenBench.json

@@ -1,5 +1,5 @@
 {
-    "Quality": 0.90,
+    "Quality": 90,
     "Throughput (token/s)": 893.68,
     "Link": "http://example.com/decode"
 }

data/decode_vLLM_Mistral-7B-v0.3_LongGenBench.json

@@ -1,5 +1,5 @@
 {
-    "Quality": 0.90,
+    "Quality": 90,
     "Throughput (token/s)": 278.95,
     "Link": "http://example.com/decode"
 }