Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
| 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]
|