Upload folder using huggingface_hub

app/draw_diagram.py

@@ -72,18 +72,14 @@ def draw(folder_name, category_name, dataset_name, metrics, cus_sort=True):
 
         chart_data_table = chart_data[['model_show', chart_data.columns[1], chart_data.columns[3]]]
 
-
         # Format numeric columns to 2 decimal places
-        chart_data_table[chart_data_table.columns[1]] = chart_data_table[chart_data_table.columns[1]].apply(lambda x: round(float(x), 3) if isinstance(float(x), (int, float)) else float(x))
-
-
+        #chart_data_table[chart_data_table.columns[1]] = chart_data_table[chart_data_table.columns[1]].apply(lambda x: round(float(x), 3) if isinstance(float(x), (int, float)) else float(x))
         cur_dataset_name = chart_data_table.columns[1]
 
 
         def highlight_first_element(x):
                 # Create a DataFrame with the same shape as the input
                 df_style = pd.DataFrame('', index=x.index, columns=x.columns)
-                
                 # Apply background color to the first element in row 0 (df[0][0])
                 df_style.iloc[0, 1] = 'background-color: #b0c1d7; color: white'
                 
@@ -118,26 +114,13 @@ def draw(folder_name, category_name, dataset_name, metrics, cus_sort=True):
                     ascending=False
                 ).reset_index(drop=True)
             
-            # styled_df = chart_data_table.style.highlight_min(
-            #     subset=[chart_data_table.columns[1]], color='yellow'
-            # )
 
-        styled_df = chart_data_table.style.apply(
+        styled_df = chart_data_table.style.format(
+            {chart_data_table.columns[1]: "{:.3f}"}
+        ).apply(
             highlight_first_element, axis=None
         )
 
-        # else:
-
-            
-
-        #     # styled_df = chart_data_table.style.highlight_max(
-        #     #     subset=[chart_data_table.columns[1]], color='yellow'
-        #     # )
-
-        #     styled_df = chart_data_table.style.apply(
-        #         highlight_first_element, axis=None
-        #     )
-
 
         st.dataframe(
                 styled_df,

app/summarization.py

@@ -102,7 +102,12 @@ def sum_table_mulit_metrix(task_name, metrics_lists: List[str]):
             df_style.iloc[0, 1] = 'background-color: #b0c1d7; color: white'
             return df_style
         
-        styled_df = chart_data_table.style.apply(
+
+        styled_df = chart_data_table.style.format(
+            {
+                chart_data_table.columns[i]: "{:.3f}" for i in range(1, len(chart_data_table.columns) - 1)
+             }
+        ).apply(
             highlight_first_element, axis=None
         )
 
@@ -119,6 +124,5 @@ def sum_table_mulit_metrix(task_name, metrics_lists: List[str]):
                 use_container_width=True
             )
             
-    #for metrics in metrics_lists:
     # Only report the last metrics
     st.markdown(f'###### Metric: {metrics_info[metrics]}')