Rthur2003 commited on
Commit
a12879f
·
1 Parent(s): 83b3658

feat: add wav2vec2 model display in build_models_md function

Browse files
Files changed (1) hide show
  1. local_demo.py +3 -0
local_demo.py CHANGED
@@ -599,6 +599,9 @@ def build_models_md() -> str:
599
  f"{result.get('accuracy', 0.0):.4f} | {result.get('f1', 0.0):.4f} |"
600
  )
601
 
 
 
 
602
  lines.extend(["", "## Secilen parametreler", ""])
603
  for name in model_names:
604
  params = training_results[name].get("selected_params", {})
 
599
  f"{result.get('accuracy', 0.0):.4f} | {result.get('f1', 0.0):.4f} |"
600
  )
601
 
602
+ if ARTIFACTS.wav2vec2_model is not None:
603
+ lines.append("| wav2vec2 (fine-tuned) | DL | — | — | — | — |")
604
+
605
  lines.extend(["", "## Secilen parametreler", ""])
606
  for name in model_names:
607
  params = training_results[name].get("selected_params", {})