Update utils_display.py
Browse files- utils_display.py +13 -1
utils_display.py
CHANGED
@@ -26,7 +26,19 @@ class AutoEvalColumn: # Auto evals column
|
|
26 |
|
27 |
|
28 |
def make_clickable_model(model_name):
|
29 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
30 |
return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'
|
31 |
|
32 |
def styled_error(error):
|
|
|
26 |
|
27 |
|
28 |
def make_clickable_model(model_name):
|
29 |
+
model_name_list = model_name.split("/")
|
30 |
+
if model_name_list[0] == "trt-llm":
|
31 |
+
link = "https://github.com/NVIDIA/TensorRT-LLM/tree/main/examples/whisper"
|
32 |
+
elif model_name_list[0] == "faster-whisper":
|
33 |
+
link = "https://github.com/guillaumekln/faster-whisper"
|
34 |
+
elif model_name_list[0] == "Whisper.cpp":
|
35 |
+
link = "https://github.com/ggerganov/whisper.cpp"
|
36 |
+
elif model_name_list[0] == "WhisperKit":
|
37 |
+
link = "https://github.com/argmaxinc/WhisperKit"
|
38 |
+
elif model_name_list[0] == "WhisperMLX":
|
39 |
+
link = "https://huggingface.co/collections/mlx-community/whisper-663256f9964fbb1177db93dc"
|
40 |
+
else:
|
41 |
+
link = f"https://huggingface.co/{model_name}"
|
42 |
return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'
|
43 |
|
44 |
def styled_error(error):
|