SimpleFrog commited on
Commit
a928657
·
verified ·
1 Parent(s): aaa9278

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ if uploaded_file is not None:
52
 
53
  # 🔹 Générer la transcription
54
  with torch.no_grad():
55
- predicted_ids = model.generate(inputs.input_features)
56
 
57
  # 🔹 Décoder la sortie
58
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
 
52
 
53
  # 🔹 Générer la transcription
54
  with torch.no_grad():
55
+ predicted_ids = model.generate(inputs.input_features, language="fr")
56
 
57
  # 🔹 Décoder la sortie
58
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]