Tlanextli commited on
Commit
81a412a
·
1 Parent(s): 4c81c7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from transformers import pipeline
5
 
6
 
7
- title = "Transcribe speech several languages"
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
  asr_pipe_audio2Text_Ge = pipeline(task="automatic-speech-recognition", model="jonatasgrosman/wav2vec2-large-xlsr-53-german")
@@ -14,9 +14,9 @@ def transcribeFile(inputlang, audio_path : str) -> str:
14
  #transcription = asr_pipe_audio2Text_Ge(audio_path)
15
  #transcription = asr_pipe_whisper(audio_path, max_new_tokens=256, generate_kwargs={"task":"transcribe"})
16
  if inputlang == "English":
17
- transcription = asr_pipe_whisper(audio_path, chunk_length_s=10, stride_length_s=(4, 2))
18
  elif inputlang == "German":
19
- transcription = asr_pipe_audio2Text_Ge(audio_path, chunk_length_s=10, stride_length_s=(4, 2))
20
  return transcription["text"]
21
 
22
  def translateAudio(audio_path):
 
4
  from transformers import pipeline
5
 
6
 
7
+ title = "Transcribe speech in several languages"
8
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
9
 
10
  asr_pipe_audio2Text_Ge = pipeline(task="automatic-speech-recognition", model="jonatasgrosman/wav2vec2-large-xlsr-53-german")
 
14
  #transcription = asr_pipe_audio2Text_Ge(audio_path)
15
  #transcription = asr_pipe_whisper(audio_path, max_new_tokens=256, generate_kwargs={"task":"transcribe"})
16
  if inputlang == "English":
17
+ transcription = asr_pipe_whisper(audio_path, chunk_length_s=25, stride_length_s=(5, 0), generate_kwargs={"task":"transcribe"})
18
  elif inputlang == "German":
19
+ transcription = asr_pipe_audio2Text_Ge(audio_path, chunk_length_s=25, stride_length_s=(5, 0))
20
  return transcription["text"]
21
 
22
  def translateAudio(audio_path):