RXTIME commited on
Commit
e36c42f
·
verified ·
1 Parent(s): bbaa68f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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: