Spaces:
Running
Running
File size: 1,126 Bytes
d8795b1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
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
|