Update app.py

app.py

@@ -23,6 +23,7 @@ demo = gr.Blocks(
     theme=gr.themes.Default(primary_hue="indigo", secondary_hue="indigo"),
 )
 with demo:
+    print("Building demo")
     gr.HTML(LOGO, elem_classes="logo")
     gr.HTML(TITLE, elem_classes="title")
     ####################### HARDWARE TABS #######################
@@ -63,64 +64,64 @@ with demo:
                         hardware_type=config.hardware_type,
                     )
                     ####################### LEADERBOARD TAB #######################
-                    # with gr.TabItem("Leaderboard 🏅", id=0):
-                    #     search_bar, columns_checkboxes, leaderboard_table = (
-                    #         create_leaderboard_table(open_llm_perf_df)
-                    #     )
-                    # with gr.TabItem("Find Your Best Model 🧭", id=1):
-                    #     lat_score_mem_plot = create_lat_score_mem_plot(
-                    #         open_llm_perf_df
-                    #     )
-                    ###################### ATTENTIONS SPEEDUP TAB #######################
-                    # with gr.TabItem("Attention 📈", id=2):
-                    #     attn_prefill_plot, attn_decode_plot = create_attn_plots(
-                    #         open_llm_perf_df
-                    #     )
-                    # ####################### KERNELS SPEEDUP TAB #######################
-                    # with gr.TabItem("Kernels 📈", id=4):
-                    #     quant_krnl_prefill_plot, quant_krnl_decode_plot = (
-                    #         create_quant_krnl_plots(llm_perf_df)
-                    #     )
+                    with gr.TabItem("Leaderboard 🏅", id=0):
+                        search_bar, columns_checkboxes, leaderboard_table = (
+                            create_leaderboard_table(open_llm_perf_df)
+                        )
+                    with gr.TabItem("Find Your Best Model 🧭", id=1):
+                        lat_score_mem_plot = create_lat_score_mem_plot(
+                            open_llm_perf_df
+                        )
+                    ##################### ATTENTIONS SPEEDUP TAB #######################
+                    with gr.TabItem("Attention 📈", id=2):
+                        attn_prefill_plot, attn_decode_plot = create_attn_plots(
+                            open_llm_perf_df
+                        )
+                    ####################### KERNELS SPEEDUP TAB #######################
+                    with gr.TabItem("Kernels 📈", id=4):
+                        quant_krnl_prefill_plot, quant_krnl_decode_plot = (
+                            create_quant_krnl_plots(llm_perf_df)
+                        )
 
-    #             ####################### CONTROL CALLBACK #######################
-    #             create_control_callback(
-    #                 filter_button,
-    #                 # inputs
-    #                 machine_value,
-    #                 subsets_value,
-    #                 backends_value,
-    #                 hardware_type_value,
-    #                 score_slider,
-    #                 memory_slider,
-    #                 backend_checkboxes,
-    #                 datatype_checkboxes,
-    #                 optimization_checkboxes,
-    #                 quantization_checkboxes,
-    #                 kernels_checkboxes,
-    #                 # interactive
-    #                 columns_checkboxes,
-    #                 search_bar,
-    #                 # outputs
-    #                 leaderboard_table,
-    #                 lat_score_mem_plot,
-    #                 # attn_prefill_plot,
-    #                 # attn_decode_plot,
-    #                 # quant_krnl_prefill_plot,
-    #                 # quant_krnl_decode_plot,
-    #             )
+                ####################### CONTROL CALLBACK #######################
+                create_control_callback(
+                    filter_button,
+                    # inputs
+                    machine_value,
+                    subsets_value,
+                    backends_value,
+                    hardware_type_value,
+                    score_slider,
+                    memory_slider,
+                    backend_checkboxes,
+                    datatype_checkboxes,
+                    optimization_checkboxes,
+                    quantization_checkboxes,
+                    kernels_checkboxes,
+                    # interactive
+                    columns_checkboxes,
+                    search_bar,
+                    # outputs
+                    leaderboard_table,
+                    lat_score_mem_plot,
+                    # attn_prefill_plot,
+                    # attn_decode_plot,
+                    # quant_krnl_prefill_plot,
+                    # quant_krnl_decode_plot,
+                )
 
-    #             create_select_callback(
-    #                 # inputs
-    #                 machine_value,
-    #                 subsets_value,
-    #                 backends_value,
-    #                 hardware_type_value,
-    #                 # interactive
-    #                 columns_checkboxes,
-    #                 search_bar,
-    #                 # outputs
-    #                 leaderboard_table,
-    #             )
+                create_select_callback(
+                    # inputs
+                    machine_value,
+                    subsets_value,
+                    backends_value,
+                    hardware_type_value,
+                    # interactive
+                    columns_checkboxes,
+                    search_bar,
+                    # outputs
+                    leaderboard_table,
+                )
 
     ####################### ABOUT TAB #######################
     with gr.TabItem("About 📖", id=len(configs)):
@@ -134,6 +135,7 @@ with demo:
                 elem_id="citation-button",
                 show_copy_button=True,
             )
+    print("Building demo end")
 
 if __name__ == "__main__":
     demo.queue().launch(ssr_mode=False)