final commit

This commit is contained in:
2021-09-19 22:06:24 +05:30
parent 840d97b058
commit 29328f46f8

View File

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