Update README.md
Browse files
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="
|
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 |
```
|