psmattas/SpeechToText
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

final commit

Browse Source
This commit is contained in:
Puranjay Savar Mattas
2021-09-19 22:06:24 +05:30
parent 840d97b058
commit 29328f46f8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -21,4 +21,4 @@ input_values = tokenizer(input_audio, return_tensors="pt").input_values
logits = model(input_values).logits logits = model(input_values).logits
predicted_ids = torch.argmax(logits, dim=-1) predicted_ids = torch.argmax(logits, dim=-1)
transcription = tokenizer.batch_decode(predicted_ids)[0] transcription = tokenizer.batch_decode(predicted_ids)[0]
print(transcription) print(transcription)