Update src/display/utils.py

src/display/utils.py

@@ -9,6 +9,10 @@ def fields(raw_class):
     return [v for k, v in raw_class.__dict__.items() if k[:2] != "__" and k[-2:] != "__"]
 
 
+def make_clickable_model(model_name: str) -> str:
+    link = f"https://huggingface.co/ivrit-ai/{model_name}"
+    return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'
+
 # These classes are for user facing column names,
 # to avoid having to change them all around the code
 # when a modif is needed
@@ -40,8 +44,14 @@ auto_eval_column_dict.append(["likes", ColumnContent, ColumnContent("Hub ❤️"
 auto_eval_column_dict.append(["still_on_hub", ColumnContent, ColumnContent("Available on the hub", "bool", False)])
 auto_eval_column_dict.append(["revision", ColumnContent, ColumnContent("Model sha", "str", False, False)])
 
-# We use make dataclass to dynamically fill the scores from Tasks
-AutoEvalColumn = make_dataclass("AutoEvalColumn", auto_eval_column_dict, frozen=True)
+@dataclass(frozen=True)
+class AutoEvalColumn: # Auto evals column
+    model = ColumnContent("Model", "markdown", False)
+    avg_wer = ColumnContent("Average WER ⬇️", "number", False)    
+
+    # TODO: The following is an example of WER for datasets, we'd want to replace those with actual relevant dataset names
+    dataset1_wer = ColumnContent("Dataset1", "number", False)
+    dataset2_wer = ColumnContent("Dataset2", "number", False)
 
 ## For the queue columns in the submission tab
 @dataclass(frozen=True)