SimpleFrog commited on
Commit
cea0ade
·
verified ·
1 Parent(s): 3f2da1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -17
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
- df.to_csv("wer_results.csv", index=False)
105
-
106
- mean_wer = df[df["WER"] != "-"]["WER"].mean()
107
- st.markdown(f"### 🎯 WER moyen (sans ponctuation) : `{mean_wer:.3f}`")
108
-
109
- # Bloc placeholder pour post-processing à venir
110
- if "Post-processing" in model_option:
111
- st.info("🛠️ Le post-processing sera ajouté prochainement ici...")
112
-
113
- if os.path.exists("wer_results.csv"):
114
- with open("wer_results.csv", "rb") as f:
115
- st.download_button(
116
- label="📥 Télécharger les résultats WER (.csv)",
117
- data=f,
118
- file_name="wer_results.csv",
119
- mime="text/csv"
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
+