Views
No views yet

1from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
2import librosa
3
4processor = AutoProcessor.from_pretrained('spow12/Visual-novel-transcriptor', language="ja", task="transcribe")
5model = AutoModelForSpeechSeq2Seq.from_pretrained('spow12/Visual-novel-transcriptor').cuda()
6model.config.forced_decoder_ids = processor.get_decoder_prompt_ids(language="ja", task="transcribe")
7
8data, _ = librosa.load(wav_path, sr=16000)
9input_features = processor(data, sampling_rate=16000, return_tensors="pt").input_features.cuda()
10predicted_ids = model.generate(input_features)
11transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
12print(transcription[0])1@misc {Visual-novel-transcriptor,
2 author = { YoungWoo Nam },
3 title = { Visual-novel-transcriptor },
4 year = 2024,
5 url = { https://huggingface.co/spow12/Visual-novel-transcriptor },
6 publisher = { Hugging Face }
7}