[Feature] Add name filtering

app.py

@@ -51,6 +51,11 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
 
             headers = ['Rank'] + check_box['essential'] + checkbox_group.value
             with gr.Row():
+                model_name = gr.Textbox(
+                    value='Input the Model Name (fuzzy, case insensitive)', 
+                    label='Model Name', 
+                    interactive=True,
+                    visible=True)
                 model_size = gr.CheckboxGroup(
                     choices=MODEL_SIZE,
                     value=MODEL_SIZE,
@@ -71,7 +76,7 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                 wrap=True,
                 visible=True)
 
-            def filter_df(fields, model_size, model_type):
+            def filter_df(fields, model_name, model_size, model_type):
                 filter_list = ['Avg Score', 'Avg Rank', 'OpenSource']
                 headers = ['Rank'] + check_box['essential'] + fields
 
@@ -86,6 +91,15 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                     df = df[df['flag']]
                     df.pop('flag')
                 df['Rank'] = list(range(1, len(df) + 1))
+                default_val = 'Input the Model Name (fuzzy, case insensitive)'
+                if model_name != default_val:
+                    print(model_name)
+                    model_name = model_name.lower()
+                    method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+                    flag = [model_name in name for name in method_names]
+                    df['TEMP_FLAG'] = flag
+                    df = df[df['TEMP_FLAG'] == True] 
+                    df.pop('TEMP_FLAG')
 
                 comp = gr.components.DataFrame(
                     value=df[headers],
@@ -97,7 +111,8 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                 return comp
 
             for cbox in [checkbox_group, model_size, model_type]:
-                cbox.change(fn=filter_df, inputs=[checkbox_group, model_size, model_type], outputs=data_component)
+                cbox.change(fn=filter_df, inputs=[checkbox_group, model_name, model_size, model_type], outputs=data_component)
+            model_name.submit(fn=filter_df, inputs=[checkbox_group, model_name, model_size, model_type], outputs=data_component)
 
         with gr.TabItem('🔍 About', elem_id='about', id=1):
             gr.Markdown(urlopen(VLMEVALKIT_README).read().decode())
@@ -122,6 +137,11 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                 s.table['Rank'] = list(range(1, len(s.table) + 1))
 
                 with gr.Row():
+                    s.model_name = gr.Textbox(
+                        value='Input the Model Name (fuzzy, case insensitive)', 
+                        label='Model Name', 
+                        interactive=True,
+                        visible=True)
                     s.model_size = gr.CheckboxGroup(
                         choices=MODEL_SIZE,
                         value=MODEL_SIZE,
@@ -143,7 +163,7 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                     visible=True)
                 s.dataset = gr.Textbox(value=dataset, label=dataset, visible=False)
 
-                def filter_df_l2(dataset_name, fields, model_size, model_type):
+                def filter_df_l2(dataset_name, fields, model_name, model_size, model_type):
                     s = structs[DATASETS.index(dataset_name)]
                     headers = ['Rank'] + s.check_box['essential'] + fields
                     df = cp.deepcopy(s.table)
@@ -155,6 +175,15 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                         df = df[df['flag']]
                         df.pop('flag')
                     df['Rank'] = list(range(1, len(df) + 1))
+                    default_val = 'Input the Model Name (fuzzy, case insensitive)'
+                    if model_name != default_val:
+                        print(model_name)
+                        model_name = model_name.lower()
+                        method_names = [x.split('</a>')[0].split('>')[-1].lower() for x in df['Method']]
+                        flag = [model_name in name for name in method_names]
+                        df['TEMP_FLAG'] = flag
+                        df = df[df['TEMP_FLAG'] == True] 
+                        df.pop('TEMP_FLAG')
 
                     comp = gr.components.DataFrame(
                         value=df[headers],
@@ -168,8 +197,12 @@ with gr.Blocks(title="Open VLM Leaderboard", head=head_style) as demo:
                 for cbox in [s.checkbox_group, s.model_size, s.model_type]:
                     cbox.change(
                         fn=filter_df_l2,
-                        inputs=[s.dataset, s.checkbox_group, s.model_size, s.model_type],
+                        inputs=[s.dataset, s.checkbox_group, s.model_name, s.model_size, s.model_type],
                         outputs=s.data_component)
+                s.model_name.submit(
+                    fn=filter_df_l2, 
+                    inputs=[s.dataset, s.checkbox_group, s.model_name, s.model_size, s.model_type],
+                    outputs=s.data_component)
 
     with gr.Row():
         with gr.Accordion('Citation', open=False):