Fix more tasks tables

app.py

@@ -12,7 +12,7 @@ hf_api = HfApi()
 
 models = list(hf_api.list_models(filter=benchmark_tag))
 
-supported_tasks = [
+SUPPORTED_TASKS = [
     'in_catalog_retrieval_zero_shot',
     'in_catalog_open_catalog',
     'in_catalog_closed_catalog',
@@ -41,7 +41,7 @@ def get_model_results(model_meta):
     metrics_meta = []
     for index in model_meta['model-index']:
         for result in index['results']:
-            if result['dataset']['type'].split('/')[0] == benchmark_user and result['dataset']['config'] in supported_tasks:
+            if result['dataset']['type'].split('/')[0] == benchmark_user and result['dataset']['config'] in SUPPORTED_TASKS:
                 metrics_dict = {metric['name']: metric['value'] for metric in result['metrics']}
                 metrics_meta += [dict(dataset=result['dataset']['type'], task=result['dataset']['config'], **metrics_dict)]
 
@@ -70,17 +70,19 @@ print(df)
 block = gr.Blocks()
 
 with block:
-    for group_name, group in df.groupby('task'):
-        gr.Markdown(f"## Task - {group_name}")
-
-    group = group.sort_values('ROC_AUC', ascending=False)
-    group['dataset'] = group['dataset'].apply(lambda x: create_model_link(x, type="datasets"))
-    gr.DataFrame(
-        group.reset_index(drop=True), 
-        datatype=['markdown', 'markdown'] + ['number'] * len(group.columns),
-        wrap=True,
-    )
-
+    for task in SUPPORTED_TASKS: 
+        group = df[df['task'] == task]
+
+        if len(group) > 0:
+            gr.Markdown(f"## Task - {task}")
+
+            group = group.sort_values('ROC_AUC', ascending=False)
+            group['dataset'] = group['dataset'].apply(lambda x: create_model_link(x, type="datasets"))
+            gr.DataFrame(
+                group.reset_index(drop=True), 
+                datatype=['markdown', 'markdown'] + ['number'] * len(group.columns),
+                wrap=True,
+            )
 
 block.launch()