|
from dataclasses import dataclass |
|
|
|
|
|
|
|
@dataclass |
|
class ColumnContent: |
|
name: str |
|
type: str |
|
|
|
def fields(raw_class): |
|
return [v for k, v in raw_class.__dict__.items() if k[:2] != "__" and k[-2:] != "__"] |
|
|
|
@dataclass(frozen=True) |
|
class AutoEvalColumn: |
|
model = ColumnContent("Model", "markdown") |
|
ap = ColumnContent("AP", "str") |
|
ap50 = ColumnContent("AP@.50", "number") |
|
ap75 = ColumnContent("AP@.75", "number") |
|
ap_small = ColumnContent("AP-S", "number") |
|
ap_medium = ColumnContent("AP-M", "number") |
|
ap_large = ColumnContent("AP-L", "number") |
|
ar_1 = ColumnContent("AR1", "number") |
|
ar_10 = ColumnContent("AR10", "number") |
|
ar_100 = ColumnContent("AR100", "number") |
|
ar_small = ColumnContent("AR-S", "number") |
|
ar_medium = ColumnContent("AR-M", "number") |
|
ar_large = ColumnContent("AR-L", "number") |
|
fps = ColumnContent("FPS(*)", "number") |
|
license = ColumnContent("hub license", "str") |
|
|
|
def make_clickable_model(model_name): |
|
link = f"https://huggingface.co/{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>' |
|
|
|
def styled_error(error): |
|
return f"<p style='color: red; font-size: 20px; text-align: center;'>{error}</p>" |
|
|
|
def styled_warning(warn): |
|
return f"<p style='color: orange; font-size: 20px; text-align: center;'>{warn}</p>" |
|
|
|
def styled_message(message): |
|
return f"<p style='color: green; font-size: 20px; text-align: center;'>{message}</p>" |
|
|
