refine

app.py

@@ -204,12 +204,12 @@ with demo:
                 if files[i] == "church":
                     name = "### [Unlearned Objects] "+" Church" 
                     csv_path = './assets/'+files[i]+'.csv'
-                elif files[i] == 'garbage':
-                    name = "### [Unlearned Objects] "+" Garbage"
-                    csv_path = './assets/'+files[i]+'.csv'
-                elif files[i] == 'tench':
-                    name = "### [Unlearned Objects] "+" Tench"
-                    csv_path = './assets/'+files[i]+'.csv'
+                # elif files[i] == 'garbage':
+                #     name = "### [Unlearned Objects] "+" Garbage"
+                #     csv_path = './assets/'+files[i]+'.csv'
+                # elif files[i] == 'tench':
+                #     name = "### [Unlearned Objects] "+" Tench"
+                #     csv_path = './assets/'+files[i]+'.csv'
                 elif files[i] == 'parachute':
                     name = "### [Unlearned Objects] "+" Parachute"
                     csv_path = './assets/'+files[i]+'.csv'
@@ -267,6 +267,70 @@ with demo:
                         ],
                         leaderboard_table,
                 )
+                    
+        with gr.TabItem("Style", elem_id="Style", id=1):
+            with gr.Row():
+                with gr.Column():
+                    with gr.Row():
+                        search_bar = gr.Textbox(
+                            placeholder=" 🔍 Search for your model (separate multiple queries with `;`) and press ENTER...",
+                            show_label=False,
+                            elem_id="search-bar",
+                        )
+                    with gr.Row():
+                        model1_column = gr.CheckboxGroup(
+                        label="Evaluation Metrics",
+                        choices=['Pre-ASR','Post-ASR','FID','CLIP-Score'],
+                        interactive=True,
+                        elem_id="column-select",
+                    )
+                   
+            for i in range(len(files)):
+                
+                if files[i] == 'vangogh':
+                    name = "### [Unlearned Style] "+" Van Gogh"
+                    csv_path = './assets/'+files[i]+'.csv'
+                
+                gr.Markdown(name)
+                df_results = load_data(csv_path)
+                df_results_init = df_results.copy()[show_columns]
+                leaderboard_table = gr.components.Dataframe(
+                value = df_results,
+                datatype = TYPES,
+                elem_id = "leaderboard-table",
+                interactive = False,
+                visible=True,
+                )
+
+            
+                hidden_leaderboard_table_for_search = gr.components.Dataframe(
+                    value=df_results_init,
+                    # value=df_results,
+                    interactive=False,
+                    visible=False,
+                )
+        
+                search_bar.submit(
+                    update_table,
+                    [   
+                        
+                        hidden_leaderboard_table_for_search,
+                        model1_column,
+                        search_bar,
+                    ],
+                    leaderboard_table,
+                )
+
+                for selector in [model1_column]:
+                    selector.change(
+                        update_table,
+                        [   
+                            hidden_leaderboard_table_for_search,
+                            model1_column,
+                            search_bar,
+                        ],
+                        leaderboard_table,
+                )