danhtran2mind
/

Vi-Whisper-Tiny-finetuning

Automatic Speech Recognition

Model card Files Files and versions

Metrics Training metrics Community

danhtran2mind commited on 26 days ago

Commit

10a752f

·

verified ·

1 Parent(s): 27c4d46

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -17,8 +17,8 @@ import librosa
 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 # Load model and processor
-processor = WhisperProcessor.from_pretrained("danhtran2mind/Vi-Whisper-tiny-finetuning")
-model = WhisperForConditionalGeneration.from_pretrained("danhtran2mind/Vi-Whisper-tiny-finetuning")
 model.config.forced_decoder_ids = None
 # Move model to GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"

 from transformers import WhisperProcessor, WhisperForConditionalGeneration
 # Load model and processor
+processor = WhisperProcessor.from_pretrained("danhtran2mind/Vi-Whisper-Tiny-finetuning")
+model = WhisperForConditionalGeneration.from_pretrained("danhtran2mind/Vi-Whisper-Tiny-finetuning")
 model.config.forced_decoder_ids = None
 # Move model to GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"