Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,10 @@ import librosa
|
|
| 39 |
# load model and processor
|
| 40 |
processor = WhisperProcessor.from_pretrained("NhutP/ViWhisper-medium")
|
| 41 |
model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-medium")
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
# load a sample
|
| 45 |
array, sampling_rate = librosa.load('path_to_audio', sr = 16000) # Load some audio sample
|
|
|
|
| 39 |
# load model and processor
|
| 40 |
processor = WhisperProcessor.from_pretrained("NhutP/ViWhisper-medium")
|
| 41 |
model = WhisperForConditionalGeneration.from_pretrained("NhutP/ViWhisper-medium")
|
| 42 |
+
|
| 43 |
+
prefix_ids = model.generation_config.forced_decoder_ids
|
| 44 |
+
model.generation_config.input_ids = prefix_ids
|
| 45 |
+
model.generation_config.forced_decoder_ids = None
|
| 46 |
|
| 47 |
# load a sample
|
| 48 |
array, sampling_rate = librosa.load('path_to_audio', sr = 16000) # Load some audio sample
|