Update app.py

app.py

@@ -94,7 +94,9 @@ with gr.Blocks() as demo:
         with gr.TabItem("Leaderboard"):
             # ...existing code...
             task_options = [col for col in df.columns if col not in ['model','hf_name','model_physical_size', 'precision']]
+            task_options.append("plot_pareto")
             with gr.Row():
+                # print pareto or not
                 selected_tasks = gr.CheckboxGroup(choices=task_options, label="Select Tasks")
             with gr.Row():
                 accuracy_plot = gr.Plot(label="Accuracy Plot")
@@ -108,6 +110,10 @@ with gr.Blocks() as demo:
             def update_outputs(selected_tasks):
                 if not selected_tasks:
                     return df[['model', 'precision']], None, None
+                plot_pareto=False
+                if "plot_pareto" in selected_tasks:
+                    plot_pareto = True
+                    selected_tasks.remove("plot_pareto")
                 filtered_df = df[['model', 'precision', 'model_physical_size','hf_name'] + selected_tasks]
                 # average accuracy of selected tasks
                 filtered_df['avg_accuracy'] = filtered_df[selected_tasks].mean(axis=1)
@@ -117,12 +123,13 @@ with gr.Blocks() as demo:
                 pareto_df = filtered_df.sort_values('model_physical_size')
                 pareto_df = pareto_df.loc[pareto_df['avg_accuracy'].cummax().drop_duplicates().index]
                 # Add Pareto frontier to line_plot
-                line_fig.add_trace(go.Scatter(
-                    x=pareto_df['model_physical_size'],
-                    y=pareto_df['avg_accuracy'],
-                    mode='lines+markers',
-                    name='Pareto Frontier'
-                ))
+                if plot_pareto:
+                    line_fig.add_trace(go.Scatter(
+                        x=pareto_df['model_physical_size'],
+                        y=pareto_df['avg_accuracy'],
+                        mode='lines+markers',
+                        name='Pareto Frontier'
+                    ))
                 
                 # set title of bar_fig
                 bar_fig.update_layout(title=f'tasks: {", ".join(selected_tasks)}')
@@ -133,14 +140,13 @@ with gr.Blocks() as demo:
                 
                 pareto_df = with_perf_df.sort_values('avg_e2e_latency')
                 pareto_df = pareto_df.loc[pareto_df['avg_accuracy'].cummax().drop_duplicates().index]
-                
-                latency_line_fig.add_trace(go.Scatter(
-                    x=pareto_df['avg_e2e_latency'],
-                    y=pareto_df['avg_accuracy'],
-                    mode='lines+markers',
-                    name='Pareto Frontier'
-                ))
-                print(with_perf_df)
+                if plot_pareto:
+                    latency_line_fig.add_trace(go.Scatter(
+                        x=pareto_df['avg_e2e_latency'],
+                        y=pareto_df['avg_accuracy'],
+                        mode='lines+markers',
+                        name='Pareto Frontier'
+                    ))
                 return with_perf_df, bar_fig, line_fig, throughput_line_fig, latency_line_fig
             
             selected_tasks.change(
@@ -158,4 +164,4 @@ with gr.Blocks() as demo:
             )
     
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)