SimpleFrog commited on
Commit
f56d7bc
·
verified ·
1 Parent(s): dc99fb5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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
- openai.api_key = api_key
123
- response = openai.ChatCompletion.create(
124
  model="gpt-4o",
125
  messages=[
126
- {"role": "system", "content": "Tu es un assistant qui corrige et ponctue des transcriptions vocales françaises sans changer le sens du texte. Répond uniquement avec le texte corrigé."},
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["content"].strip()
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