Upload folder using huggingface_hub

app/draw_diagram.py

@@ -22,7 +22,7 @@ info_df = pd.read_excel(path)
 #     except:
 #         pass
 
-def draw(folder_name, category_name, dataset_name, metrics):
+def draw(folder_name, category_name, dataset_name, metrics, cus_sort=True):
     
     folder = f"./results/{metrics}/"
 
@@ -59,7 +59,7 @@ def draw(folder_name, category_name, dataset_name, metrics):
     
     chart_data = chart_data[chart_data['model_show'].isin(models)]
     
-    chart_data = chart_data.sort_values(by=[new_dataset_name], ascending=True).dropna(axis=0)
+    chart_data = chart_data.sort_values(by=[new_dataset_name], ascending=cus_sort).dropna(axis=0)
 
     # import pdb
     # pdb.set_trace()
@@ -133,11 +133,13 @@ def draw(folder_name, category_name, dataset_name, metrics):
         # import pdb
         # pdb.set_trace()
 
+        chart_data_table = chart_data[['model_show', chart_data.columns[1], chart_data.columns[3]]]
+
         st.dataframe(
-                chart_data,
+                chart_data_table,
                 column_config={
-                    'model_show': "Model",
-                    chart_data.columns[1]: {'alignment': 'center'},
+                    'model_show': 'Model',
+                    chart_data_table.columns[1]: {'alignment': 'center'},
                     "model_link": st.column_config.LinkColumn(
                         "Model Link",
                         # # # help="",

app/pages.py

@@ -128,7 +128,7 @@ def asr():
 
     if filter_1:
         dataset_contents(asr_datsets[filter_1], metrics['wer'])
-        draw('su', 'ASR', filter_1, 'wer')
+        draw('su', 'ASR', filter_1, 'wer', cus_sort=True)
     # else:
     #     draw('su', 'ASR', 'LibriSpeech-Test-Clean', 'wer')