SimpleFrog commited on
Commit
3914468
·
verified ·
1 Parent(s): adf9bbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -66,11 +66,11 @@ if uploaded_file is not None:
66
  audio = np.expand_dims(audio, axis=0) # Ajouter une dimension batch
67
 
68
  # 🔹 Préparer les entrées pour Whisper
69
- inputs = processor(audio, sampling_rate=16000, return_tensors="pt")
70
 
71
  # 🔹 Générer la transcription
72
  with torch.no_grad():
73
- predicted_ids = model.generate(inputs.input_features, language="fr")
74
 
75
  # 🔹 Décoder la sortie
76
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
 
66
  audio = np.expand_dims(audio, axis=0) # Ajouter une dimension batch
67
 
68
  # 🔹 Préparer les entrées pour Whisper
69
+ inputs = processor(audio, sampling_rate=16000, return_tensors="pt", language="fr", task="transcribe")
70
 
71
  # 🔹 Générer la transcription
72
  with torch.no_grad():
73
+ predicted_ids = model.generate(inputs.input_features)
74
 
75
  # 🔹 Décoder la sortie
76
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]