src/formatting.py

def model_hyperlink(link, 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>'


MODEL_URLS = {
    "chronos_tiny": "amazon/chronos-t5-tiny",
    "chronos_mini": "amazon/chronos-t5-mini",
    "chronos_small": "amazon/chronos-t5-small",
    "chronos_base": "amazon/chronos-t5-base",
    "chronos_large": "amazon/chronos-t5-large",
    "chronos_bolt_tiny": "amazon/chronos-bolt-tiny",
    "chronos_bolt_mini": "amazon/chronos-bolt-mini",
    "chronos_bolt_small": "amazon/chronos-bolt-small",
    "chronos_bolt_base": "amazon/chronos-bolt-base",
    "moirai_large": "Salesforce/moirai-1.1-R-large",
    "moirai_base": "Salesforce/moirai-1.1-R-base",
    "moirai_small": "Salesforce/moirai-1.1-R-small",
    "timesfm": "google/timesfm-1.0-200m",
}


def make_clickable_model(model_name):
    if model_name in MODEL_URLS:
        model_path = MODEL_URLS.get(model_name)
        link = f"https://huggingface.co/{model_path}"
        return model_hyperlink(link, model_name)
    else:
        return model_name