vasista22 commited on
Commit
883620c
·
1 Parent(s): 337cad7

inference_code_snippet_added

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -51,8 +51,8 @@ In order to infer a single audio file using this model, the following code snipp
51
  >>> device = "cuda:0" if torch.cuda.is_available() else "cpu"
52
 
53
  >>> transcribe = pipeline(task="automatic-speech-recognition", model="vasista22/whisper-kannada-base", chunk_length_s=30, device=device)
54
-
55
  >>> transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="kn", task="transcribe")
 
56
  >>> print('Transcription: ', transcribe(audio)["text"])
57
  ```
58
 
@@ -83,6 +83,6 @@ The following hyperparameters were used during training:
83
  - mixed_precision_training: True
84
 
85
  ## Acknowledgement
86
- This work was done at [Speech Lab, IITM](https://asr.iitm.ac.in/).
87
 
88
  The compute resources for this work were funded by "Bhashini: National Language translation Mission" project of the Ministry of Electronics and Information Technology (MeitY), Government of India.
 
51
  >>> device = "cuda:0" if torch.cuda.is_available() else "cpu"
52
 
53
  >>> transcribe = pipeline(task="automatic-speech-recognition", model="vasista22/whisper-kannada-base", chunk_length_s=30, device=device)
 
54
  >>> transcribe.model.config.forced_decoder_ids = transcribe.tokenizer.get_decoder_prompt_ids(language="kn", task="transcribe")
55
+
56
  >>> print('Transcription: ', transcribe(audio)["text"])
57
  ```
58
 
 
83
  - mixed_precision_training: True
84
 
85
  ## Acknowledgement
86
+ This work was done at [Speech Lab, IIT Madras](https://asr.iitm.ac.in/).
87
 
88
  The compute resources for this work were funded by "Bhashini: National Language translation Mission" project of the Ministry of Electronics and Information Technology (MeitY), Government of India.