Rthur2003 commited on
Commit
fcf0dbd
·
1 Parent(s): 3ad0b90

feat: add audio path parameter to _build_model_table_html and include wav2vec2 predictions

Browse files
Files changed (1) hide show
  1. 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>',