Update app.py

app.py

@@ -2,17 +2,14 @@ from utils import *
 
 global data_component
 
-
 def update_table(query, min_size, max_size):
     df = get_df()
     filtered_df = search_and_filter_models(df, query, min_size, max_size)
     return filtered_df
 
-
 with gr.Blocks() as block:
-    gr.Markdown(
-        LEADERBOARD_INTRODUCTION
-    )
+    gr.Markdown(LEADERBOARD_INTRODUCTION)
+    
     with gr.Tabs(elem_classes="tab-buttons") as tabs:
         # Table 1
         with gr.TabItem("📊 MMLU-Pro", elem_id="qa-tab-table1", id=1):
@@ -24,9 +21,7 @@ with gr.Blocks() as block:
                         elem_id="citation-button",
                         lines=10,
                     )
-            gr.Markdown(
-                TABLE_INTRODUCTION
-            )
+            gr.Markdown(TABLE_INTRODUCTION)
 
             with gr.Row():
                 search_bar = gr.Textbox(
@@ -34,12 +29,9 @@ with gr.Blocks() as block:
                     show_label=False,
                     elem_id="search-bar"
                 )
+            
             df = get_df()
-            # min_size, max_size = get_size_range(df)
-
-            numeric_sizes = df[df['Model Size(B)'].apply(lambda x: isinstance(x, (int, float)))]['Model Size(B)']
-            min_size = numeric_sizes.min() if not numeric_sizes.empty else 0
-            max_size = numeric_sizes.max() if not numeric_sizes.empty else 1000
+            min_size, max_size = get_size_range(df)
 
             with gr.Row():
                 min_size_slider = gr.Slider(
@@ -66,6 +58,7 @@ with gr.Blocks() as block:
                 visible=True,
                 height=1000
             )
+            
             refresh_button = gr.Button("Refresh")
             search_bar.change(fn=update_table, inputs=[search_bar, min_size_slider, max_size_slider], outputs=data_component)
             min_size_slider.change(fn=update_table, inputs=[search_bar, min_size_slider, max_size_slider], outputs=data_component)
@@ -81,6 +74,4 @@ with gr.Blocks() as block:
             with gr.Row():
                 gr.Markdown(SUBMIT_INTRODUCTION, elem_classes="markdown-text")
 
-
-block.launch(share=True)
-
+block.launch(share=True)