Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,6 +19,7 @@ import tempfile
|
|
19 |
from huggingface_hub import snapshot_download
|
20 |
from transformers import pipeline
|
21 |
import openai
|
|
|
22 |
|
23 |
|
24 |
st.title("📊 Évaluation WER d'un modèle Whisper")
|
@@ -119,17 +120,17 @@ if start_eval:
|
|
119 |
|
120 |
#fonction process GPT4o
|
121 |
def postprocess_with_gpt4o(text, api_key):
|
122 |
-
|
123 |
-
response =
|
124 |
model="gpt-4o",
|
125 |
messages=[
|
126 |
-
{"role": "system", "content": "Tu es
|
127 |
{"role": "user", "content": f"Corrige ce texte : {text}"}
|
128 |
],
|
129 |
temperature=0.3,
|
130 |
max_tokens=512
|
131 |
)
|
132 |
-
return response.choices[0].message
|
133 |
|
134 |
|
135 |
# 🔹 Préparer WER metric
|
|
|
19 |
from huggingface_hub import snapshot_download
|
20 |
from transformers import pipeline
|
21 |
import openai
|
22 |
+
from openai import OpenAI
|
23 |
|
24 |
|
25 |
st.title("📊 Évaluation WER d'un modèle Whisper")
|
|
|
120 |
|
121 |
#fonction process GPT4o
|
122 |
def postprocess_with_gpt4o(text, api_key):
|
123 |
+
client = OpenAI(api_key=api_key)
|
124 |
+
response = client.chat.completions.create(
|
125 |
model="gpt-4o",
|
126 |
messages=[
|
127 |
+
{"role": "system", "content": "Tu es CorrecteurAI, une AI française qui permet de corriger les erreurs de saisie vocal. La translation d'un enregistrement audio tiré d'une inspection détaillé de pont t'es envoyé et tu renvoies le texte identique mais avec les éventuelles corrections si des erreurs sont détectés. Le texte peut comprendre du vocabulaire technique associé aux ouvrages d'art. Renvoies uniquement le texte corrigé en français et sans autre commentaire."},
|
128 |
{"role": "user", "content": f"Corrige ce texte : {text}"}
|
129 |
],
|
130 |
temperature=0.3,
|
131 |
max_tokens=512
|
132 |
)
|
133 |
+
return response.choices[0].message.content.strip()
|
134 |
|
135 |
|
136 |
# 🔹 Préparer WER metric
|