Upload folder using huggingface_hub

app/draw_diagram.py

@@ -33,15 +33,20 @@ def draw(folder_name, category_name, dataset_name, metrics):
     
     data_path = f'{folder}/{category_name.lower()}.csv'
     chart_data = pd.read_csv(data_path).round(3)
-
-    # if sorted == 'Ascending':
-    #     ascend = True 
-    # else:
-    #     ascend = False
-
     new_dataset_name = dataset_name.replace('-', '_').lower()
     chart_data = chart_data[['Model', new_dataset_name]]
     
+    st.markdown("""
+                <style>
+                    .stMultiSelect [data-baseweb=select] span{
+                        max-width: 800px;
+                        font-size: 0.9rem;
+                    }
+                </style>
+                """, unsafe_allow_html=True)
+    models = st.multiselect("Please choose the model", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
+    chart_data = chart_data[chart_data['Model'].isin(models)]
+    
     chart_data = chart_data.sort_values(by=[new_dataset_name], ascending=True).dropna(axis=0)
 
     if len(chart_data) == 0:
@@ -63,7 +68,7 @@ def draw(folder_name, category_name, dataset_name, metrics):
         "xAxis": [
             {
                 "type": "category",
-                "boundaryGap": False,
+                "boundaryGap": True,
                 "triggerEvent": True,
                 "data": chart_data['Model'].tolist(),
             }
@@ -71,6 +76,7 @@ def draw(folder_name, category_name, dataset_name, metrics):
         "yAxis": [{"type": "value", 
                     "min": min_value,
                     "max": max_value, 
+                    "boundaryGap": True
                     # "splitNumber": 10
                     }],
         "series": [{
@@ -111,5 +117,8 @@ def draw(folder_name, category_name, dataset_name, metrics):
     '''
     show samples
     '''
-    show_examples(category_name, dataset_name, chart_data['Model'].tolist())
+    if dataset_name in ['Earnings21-Test', 'Earnings22-Test', 'Tedlium3-Longform-Test']:
+        pass
+    else:
+        show_examples(category_name, dataset_name, chart_data['Model'].tolist())
     

app/show_examples.py

@@ -5,6 +5,7 @@ import numpy as np
 def show_examples(category_name, dataset_name, model_lists):
     st.divider()
     sample_folder = f"./examples/{category_name}/{dataset_name}"
+    
     dataset = datasets.load_from_disk(sample_folder)
 
     for index in range(len(dataset)):