xnevar00 commited on
Commit
03d6080
1 Parent(s): 0dc6445

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ from transformers import pipeline
29
  audio = "path/to/audio.xx"
30
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
31
 
32
- transcribe = pipeline(task="automatic-speech-recognition", model="whisper_full", chunk_length_s=30, device=device)
33
  transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(task="transcribe", language="czech")
34
  print('Transcription:', transcribe(audio)["text"])
35
  ```
 
29
  audio = "path/to/audio.xx"
30
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
31
 
32
+ transcribe = pipeline(task="automatic-speech-recognition", model="BUT-FIT/whisper-ATC-czech-full", chunk_length_s=30, device=device)
33
  transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(task="transcribe", language="czech")
34
  print('Transcription:', transcribe(audio)["text"])
35
  ```