Lenylvt commited on
Commit
5f9e84f
1 Parent(s): c39509c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -14
app.py CHANGED
@@ -15,21 +15,14 @@ client = Client(API_URL)
15
  def transcrire_audio(chemin_audio, task="transcription", return_timestamps=True):
16
  """Fonction pour transcrire un fichier audio en utilisant le point d'API Whisper-JAX."""
17
  with open(chemin_audio, "rb") as file:
18
- files = {
19
- "file": file
20
- }
21
- data = {
22
- "task": task,
23
- "return_timestamps": return_timestamps
24
- }
25
- # Appel synchrone à la méthode de prédiction, en utilisant `files` pour les données de fichier
26
- texte, duree = client.predict(
27
- files=files,
28
- data=data,
29
  api_name="/predict_1" # Assurez-vous que c'est le bon endpoint
30
  )
31
- return texte, duree
32
-
33
 
34
  # Widget Streamlit pour télécharger un fichier audio
35
  fichier_telecharge = st.file_uploader("Choisissez un fichier audio", type=['mp3', 'wav', 'ogg'])
@@ -64,4 +57,4 @@ def remove_timestamps(texte):
64
  motif = r"\[\d{2}:\d{2}:\d{2}\.\d{3} -> \d{2}:\d{2}:\d{2}\.\d{3}\]\s*"
65
  # Remplacer les motifs correspondants par une chaîne vide
66
  texte_nettoye = re.sub(motif, "", texte)
67
- return texte_nettoye
 
15
  def transcrire_audio(chemin_audio, task="transcription", return_timestamps=True):
16
  """Fonction pour transcrire un fichier audio en utilisant le point d'API Whisper-JAX."""
17
  with open(chemin_audio, "rb") as file:
18
+ # Préparation de la requête
19
+ response = client.predict(
20
+ file,
21
+ task,
22
+ return_timestamps,
 
 
 
 
 
 
23
  api_name="/predict_1" # Assurez-vous que c'est le bon endpoint
24
  )
25
+ return response[0], response[1] # Ajustez selon la structure de la réponse retournée par l'API
 
26
 
27
  # Widget Streamlit pour télécharger un fichier audio
28
  fichier_telecharge = st.file_uploader("Choisissez un fichier audio", type=['mp3', 'wav', 'ogg'])
 
57
  motif = r"\[\d{2}:\d{2}:\d{2}\.\d{3} -> \d{2}:\d{2}:\d{2}\.\d{3}\]\s*"
58
  # Remplacer les motifs correspondants par une chaîne vide
59
  texte_nettoye = re.sub(motif, "", texte)
60
+ return texte_nettoye