updated accuracy pattern
Browse files
app.py
CHANGED
|
@@ -105,7 +105,7 @@ def extract_metric(model_card_content, task):
|
|
| 105 |
Extract the metric value from the models' model card
|
| 106 |
:param model_card_content: model card content
|
| 107 |
"""
|
| 108 |
-
accuracy_pattern = r"Accuracy: (\d+\.\d+)"
|
| 109 |
wer_pattern = r"Wer: (\d+\.\d+)"
|
| 110 |
|
| 111 |
if task == "audio-classification":
|
|
|
|
| 105 |
Extract the metric value from the models' model card
|
| 106 |
:param model_card_content: model card content
|
| 107 |
"""
|
| 108 |
+
accuracy_pattern = r"(?:Accuracy|eval_accuracy): (\d+\.\d+)"
|
| 109 |
wer_pattern = r"Wer: (\d+\.\d+)"
|
| 110 |
|
| 111 |
if task == "audio-classification":
|