Views
No views yet
| Output wav2vec2 | Output spell correcor |
| ывсем привет выныканалетоп армии и это двадцать пятый день спец операций на украине ет самый главной новости российские военные ракетами кинжалы калибр уничтожили крупную военную топливную базу украины ракетным ударом по населенному пункту под жетамиром уничтжены более стаукраинских военных в две тысячи двадцать втором году | Всем привет! Вы в курсе новостей от армии. И это 25 день спецопераций на Украине. Есть самые главные новости. Российские военные ракетами «Кинжалы» и «Кинжалы» калибра уничтожили крупную военную топливную базу Украины. Ракетным ударом по населенному пункту под Жетамиром уничтожены более ста украинских военных в 2022г. |
1from transformers import AutoModelForCTC, Wav2Vec2Processor
2model = AutoModelForCTC.from_pretrained("UrukHan/wav2vec2-russian")
3processor = Wav2Vec2Processor.from_pretrained("UrukHan/wav2vec2-russian")
4def map_to_result(batch):
5 with torch.no_grad():
6 input_values = torch.tensor(batch["input_values"]).unsqueeze(0) #, device="cuda"
7 logits = model(input_values).logits
8 pred_ids = torch.argmax(logits, dim=-1)
9 batch = processor.batch_decode(pred_ids)[0]
10 return batch
11
12 map_to_result()