Upload folder using huggingface_hub

app/draw_diagram.py

@@ -75,7 +75,17 @@ def draw_cross_lingual(category_one, category_two, sort, sorted):
     subtitle = ''
     data_path = f'{folder}/{category_one}/{category_two}.csv'
     chart_data = pd.read_csv(data_path).dropna(axis='columns').round(3)
-
+    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 models", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
+    chart_data = chart_data[chart_data['Model'].isin(models)]
+    
     if sorted == 'Ascending':
         ascend = True 
     else:
@@ -86,15 +96,7 @@ def draw_cross_lingual(category_one, category_two, sort, sorted):
     min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1) 
     max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1) 
     
-    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 models", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
+    
 
     if category_two in ['cross_mmlu', 'cross_logiqa']:  
         # print(category_two)
@@ -121,7 +123,7 @@ def draw_cross_lingual(category_one, category_two, sort, sorted):
                     "type": "category",
                     "boundaryGap": True,
                     "triggerEvent": True,
-                    "data": models,
+                    "data": chart_data['Model'].tolist(),
                 }
             ],
             "yAxis": [{"type": "value", 
@@ -219,7 +221,7 @@ def draw_cross_lingual(category_one, category_two, sort, sorted):
                 {
                     "type": "category",
                     "boundaryGap": True,
-                    "data": models,
+                    "data": chart_data['Model'].tolist(),
                 }
             ],
             "yAxis": [{"type": "value",
@@ -343,6 +345,7 @@ def draw_only_acc(folder_name, category_one, category_two, sorted):
                 </style>
                 """, unsafe_allow_html=True)
     models = st.multiselect("Please choose the models", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
+    chart_data = chart_data[chart_data['Model'].isin(models)]
     
     if sorted == 'Ascending':
         ascend = True 
@@ -369,7 +372,7 @@ def draw_only_acc(folder_name, category_one, category_two, sorted):
                 "type": "category",
                 "boundaryGap": True,
                 "triggerEvent": True,
-                "data": models,
+                "data": chart_data['Model'].tolist(),
             }
         ],
         "yAxis": [{"type": "value", 
@@ -424,17 +427,6 @@ def draw_flores_translation(category_one, category_two, sorted):
     
     data_path = f'{folder}/{category_one}/{subtitle}.csv'
     chart_data = pd.read_csv(data_path).round(3)
-
-    if sorted == 'Ascending':
-        ascend = True 
-    else:
-        ascend = False
-    
-    chart_data = chart_data.sort_values(by=['BLEU'], ascending=ascend)
-
-    min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1) 
-    max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1) 
-    
     st.markdown("""
                 <style>
                     .stMultiSelect [data-baseweb=select] span{
@@ -444,7 +436,18 @@ def draw_flores_translation(category_one, category_two, sorted):
                 </style>
                 """, unsafe_allow_html=True)
     models = st.multiselect("Please choose the models", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
+    chart_data = chart_data[chart_data['Model'].isin(models)]
+    
+    if sorted == 'Ascending':
+        ascend = True 
+    else:
+        ascend = False
+    
+    chart_data = chart_data.sort_values(by=['BLEU'], ascending=ascend)
 
+    min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1) 
+    max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1) 
+    
     options = {
         "title": {"text": f"{category_two}"},
         "tooltip": {
@@ -460,7 +463,7 @@ def draw_flores_translation(category_one, category_two, sorted):
                 "type": "category",
                 "boundaryGap": True,
                 "triggerEvent": True,
-                "data": models,
+                "data": chart_data['Model'].tolist(),
             }
         ],
         "yAxis": [{"type": "value", 
@@ -523,7 +526,8 @@ def draw_dialogue(category_one, category_two, sort, sorted):
                 </style>
                 """, unsafe_allow_html=True)
     models = st.multiselect("Please choose the models", chart_data['Model'].tolist(), default = chart_data['Model'].tolist())
-
+    chart_data = chart_data[chart_data['Model'].isin(models)]
+    
     if sorted == 'Ascending':
         ascend = True 
     else:
@@ -534,6 +538,7 @@ def draw_dialogue(category_one, category_two, sort, sorted):
     min_value = round(chart_data.iloc[:, 1::].min().min() - 0.1, 1) 
     max_value = round(chart_data.iloc[:, 1::].max().max() + 0.1, 1) 
     
+
     options = {}
     if category_two in ['SAMSum', 'DialogSum']:
         options = {
@@ -551,7 +556,7 @@ def draw_dialogue(category_one, category_two, sort, sorted):
                     "type": "category",
                     "boundaryGap": True,
                     "triggerEvent": True,
-                    "data": models,
+                    "data": chart_data['Model'].tolist(),
                 }
             ],
             "yAxis": [{"type": "value", 
@@ -601,7 +606,7 @@ def draw_dialogue(category_one, category_two, sort, sorted):
                     "type": "category",
                     "boundaryGap": True,
                     "triggerEvent": True,
-                    "data": models,
+                    "data": chart_data['Model'].tolist(),
                 }
             ],
             "yAxis": [{"type": "value",