Update app.py

app.py

@@ -7,8 +7,6 @@ import tempfile
 
 from constants import *
 from huggingface_hub import Repository
-print(gr.__version__)
-print(pd.__version__)
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 global data_component, filter_component
@@ -90,6 +88,28 @@ def get_all_df():
     df = df.sort_values(by="Avg", ascending=False)
     return df
 
+def on_filter_model_size_method_change(selected_columns):
+    updated_data = get_all_df()
+
+    # columns:
+    selected_columns = [item for item in TASK_INFO if item in selected_columns]
+    present_columns = MODEL_INFO + selected_columns
+    print("selected_columns",'|'.join(selected_columns))
+    updated_data = updated_data[present_columns]
+    updated_data = updated_data.sort_values(by=selected_columns[0], ascending=False)
+    updated_headers = present_columns
+    update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES.index(x)] for x in updated_headers]
+    print(updated_data,present_columns,update_datatype)
+    filter_component = gr.components.Dataframe(
+        value=updated_data, 
+        headers=updated_headers,
+        type="pandas", 
+        datatype=update_datatype,
+        interactive=False,
+        visible=True,
+        )
+
+    return filter_component.value
 
 block = gr.Blocks()
 
@@ -128,29 +148,7 @@ with block:
                 interactive=False,
                 visible=True,
                 )
-    
-            def on_filter_model_size_method_change(selected_columns):
-                updated_data = get_all_df()
-
-                # columns:
-                selected_columns = [item for item in TASK_INFO if item in selected_columns]
-                present_columns = MODEL_INFO + selected_columns
-                print("selected_columns",'|'.join(selected_columns))
-                updated_data = updated_data[present_columns]
-                updated_data = updated_data.sort_values(by=selected_columns[0], ascending=False)
-                updated_headers = present_columns
-                update_datatype = [DATA_TITILE_TYPE[COLUMN_NAMES.index(x)] for x in updated_headers]
-
-                filter_component = gr.components.Dataframe(
-                    value=updated_data, 
-                    headers=updated_headers,
-                    type="pandas", 
-                    datatype=update_datatype,
-                    interactive=False,
-                    visible=True,
-                    )
-        
-                return filter_component.value
+
 
             checkbox_group.change(fn=on_filter_model_size_method_change, inputs=[ checkbox_group], outputs=data_component)