Spaces:
Running
Running
feat: add audio path parameter to _build_model_table_html and include wav2vec2 predictions
Browse files- local_demo.py +7 -0
local_demo.py
CHANGED
|
@@ -407,6 +407,7 @@ def _is_dl_wrapper(model: Any) -> bool:
|
|
| 407 |
def _build_model_table_html(
|
| 408 |
selected_model_name: str,
|
| 409 |
feature_vector: np.ndarray,
|
|
|
|
| 410 |
) -> str:
|
| 411 |
scaled = ARTIFACTS.scaler.transform(feature_vector)
|
| 412 |
scored_rows: list[tuple[str, float]] = []
|
|
@@ -422,6 +423,12 @@ def _build_model_table_html(
|
|
| 422 |
continue
|
| 423 |
scored_rows.append((name, probability))
|
| 424 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 425 |
scored_rows.sort(key=lambda item: item[1], reverse=True)
|
| 426 |
parts = [
|
| 427 |
'<section class="panel-card"><div class="panel-card__title">Model karsilastirmasi</div>',
|
|
|
|
| 407 |
def _build_model_table_html(
|
| 408 |
selected_model_name: str,
|
| 409 |
feature_vector: np.ndarray,
|
| 410 |
+
audio_path: str | None = None,
|
| 411 |
) -> str:
|
| 412 |
scaled = ARTIFACTS.scaler.transform(feature_vector)
|
| 413 |
scored_rows: list[tuple[str, float]] = []
|
|
|
|
| 423 |
continue
|
| 424 |
scored_rows.append((name, probability))
|
| 425 |
|
| 426 |
+
# wav2vec2 row
|
| 427 |
+
if ARTIFACTS.wav2vec2_model is not None and audio_path is not None:
|
| 428 |
+
w2v_prob = _wav2vec2_predict(ARTIFACTS.wav2vec2_model, audio_path)
|
| 429 |
+
if w2v_prob is not None:
|
| 430 |
+
scored_rows.append(("wav2vec2", w2v_prob))
|
| 431 |
+
|
| 432 |
scored_rows.sort(key=lambda item: item[1], reverse=True)
|
| 433 |
parts = [
|
| 434 |
'<section class="panel-card"><div class="panel-card__title">Model karsilastirmasi</div>',
|