patrickvonplaten
commited on
Commit
•
7ef592a
1
Parent(s):
e479cfb
Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ logits = model(input_values).logits
|
|
45 |
|
46 |
-prediction_ids = torch.argmax(logits, dim=-1)
|
47 |
-transcription = processor.batch_decode(prediction_ids)
|
48 |
-
+transcription = processor.batch_decode(logits)
|
49 |
|
50 |
print(transcription)
|
51 |
```
|
|
|
45 |
|
46 |
-prediction_ids = torch.argmax(logits, dim=-1)
|
47 |
-transcription = processor.batch_decode(prediction_ids)
|
48 |
+
+transcription = processor.batch_decode(logits).text
|
49 |
|
50 |
print(transcription)
|
51 |
```
|