Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ import torch
|
|
15 |
import librosa
|
16 |
import numpy as np
|
17 |
import evaluate
|
|
|
18 |
|
19 |
st.title("📊 Évaluation WER d'un modèle Whisper")
|
20 |
st.markdown("Ce Space permet d'évaluer la performance WER d'un modèle Whisper sur un dataset audio.")
|
@@ -101,20 +102,26 @@ if start_eval:
|
|
101 |
|
102 |
# 🔹 Générer le tableau de résultats
|
103 |
df = pd.DataFrame(results)
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
|
113 |
-
if
|
114 |
-
|
115 |
-
|
116 |
-
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
import librosa
|
16 |
import numpy as np
|
17 |
import evaluate
|
18 |
+
import tempfile
|
19 |
|
20 |
st.title("📊 Évaluation WER d'un modèle Whisper")
|
21 |
st.markdown("Ce Space permet d'évaluer la performance WER d'un modèle Whisper sur un dataset audio.")
|
|
|
102 |
|
103 |
# 🔹 Générer le tableau de résultats
|
104 |
df = pd.DataFrame(results)
|
105 |
+
|
106 |
+
# 🔹 Créer un fichier temporaire pour le CSV
|
107 |
+
with tempfile.NamedTemporaryFile(mode="w+", delete=False, suffix=".csv") as tmp_csv:
|
108 |
+
df.to_csv(tmp_csv.name, index=False)
|
109 |
+
|
110 |
+
mean_wer = df[df["WER"] != "-"]["WER"].mean()
|
111 |
+
st.markdown(f"### 🎯 WER moyen (sans ponctuation) : `{mean_wer:.3f}`")
|
112 |
+
|
113 |
+
# Bloc placeholder pour post-processing à venir
|
114 |
+
if "Post-processing" in model_option:
|
115 |
+
st.info("🛠️ Le post-processing sera ajouté prochainement ici...")
|
116 |
+
|
117 |
+
|
118 |
+
# 🔹 Bouton de téléchargement
|
119 |
+
with open(tmp_csv.name, "rb") as f:
|
120 |
+
st.download_button(
|
121 |
+
label="📥 Télécharger les résultats WER (.csv)",
|
122 |
+
data=f,
|
123 |
+
file_name="wer_results.csv",
|
124 |
+
mime="text/csv"
|
125 |
+
)
|
126 |
+
|
127 |
+
|