adding filtering by dtype and updating dataset

src/display.py

@@ -24,7 +24,9 @@ def display_app():
     categories_display = ["ARC", "GSM8K", "TruthfulQA", "Winogrande", "HellaSwag", "MMLU", "Average"]
 
     st.markdown("## Leaderboard")
-    sort_selection = st.selectbox(label = "Sort by:", options = list(dataframe.columns), index = 7)
+    sort_selection = st.selectbox(label = "Sort by:", options = list(dataframe.columns.difference(["model_dtype"])), index = 1)
+    d_type_options = ["all", "torch.bfloat16", "torch.float16", "4bit", "8bit"]
+    d_type = st.radio(label = "Filter by dtype", options = d_type_options, index = 0, horizontal = True)
     number_of_row = st.sidebar.number_input("Number of top rows to display", min_value=100, max_value=500, value="min", step=100)
     ascending = True
     
@@ -74,6 +76,8 @@ def display_app():
 
     valid_categories = validate_categories(ordering_metrics)
     dataframe = sort_by(dataframe=dataframe, column_name=sort_selection, ascending= ascending)
+    if d_type != "all":
+        dataframe = dataframe[dataframe["model_dtype"] == d_type]
     dataframe_display = dataframe.copy()  
     dataframe_display = show_dataframe_top(number_of_row,dataframe_display)    
     dataframe_display[["ARC", "HellaSwag", "TruthfulQA", "Winogrande", "GSM8K" ,"MMLU", "Average"]] = dataframe[["ARC", "HellaSwag", "TruthfulQA", "Winogrande", "GSM8K" ,"MMLU", "Average"]].astype(float)
@@ -173,6 +177,7 @@ def display_app():
                                    round(model_list[i]["HellaSwag"],2),
                                    round(model_list[i]["MMLU"],2)
                                    ))
+                st.markdown("**dtype:** %s" % model_list[i]["model_dtype"])
     elif len(model_list) == 1:
         st.markdown("**Model name:**   [%s](https://huggingface.co/%s)" % (model_list[0]["model_name"] , model_list[0]["model_name"]))
         st.markdown("**Results:**")
@@ -192,7 +197,8 @@ def display_app():
                                    round(model_list[0]["HellaSwag"],2),
                                    round(model_list[0]["MMLU"],2)
                                    ))
+        st.markdown("**dtype:** %s" % model_list[0]["model_dtype"])
         st.markdown("For more details, hover over the radar chart.")
     else:
         st.markdown("**Model name:**   %s" % model_name)
-        st.markdown("For more details, select the model.")
+        st.markdown("For more details, select the first model in the list/leaderboard.")