Update app.py
Browse files
app.py
CHANGED
@@ -46,8 +46,8 @@ def transcribe_audio(audio_path: str) -> str:
|
|
46 |
model = WhisperForConditionalGeneration.from_pretrained(WHISPER_MODEL)
|
47 |
input_features = processor(audio, sampling_rate=16000, return_tensors="pt").input_features
|
48 |
|
49 |
-
# Gerar transcrição
|
50 |
-
predicted_ids = model.generate(input_features)
|
51 |
transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
|
52 |
return transcription
|
53 |
except Exception as e:
|
|
|
46 |
model = WhisperForConditionalGeneration.from_pretrained(WHISPER_MODEL)
|
47 |
input_features = processor(audio, sampling_rate=16000, return_tensors="pt").input_features
|
48 |
|
49 |
+
# Gerar transcrição (forçando o idioma para inglês)
|
50 |
+
predicted_ids = model.generate(input_features, language="en", task="transcribe")
|
51 |
transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
|
52 |
return transcription
|
53 |
except Exception as e:
|