Commit
·
f9ab6fe
1
Parent(s):
b2d9405
Update README.md
Browse files
README.md
CHANGED
|
@@ -78,8 +78,6 @@ print(prediction[0])
|
|
| 78 |
## Evaluation
|
| 79 |
For the evaluation use the code below:
|
| 80 |
```python
|
| 81 |
-
pip install datasets
|
| 82 |
-
pip install transformers
|
| 83 |
import torch
|
| 84 |
import torchaudio
|
| 85 |
import librosa
|
|
@@ -114,7 +112,12 @@ def predict(batch):
|
|
| 114 |
batch["prediction"] = processor.batch_decode(logits.numpy()).text
|
| 115 |
return batch
|
| 116 |
|
| 117 |
-
dataset = load_dataset("csv", data_files={"test":"path/to/your.csv"}, delimiter=",")["test"]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 118 |
dataset = dataset.map(speech_file_to_array_fn)
|
| 119 |
|
| 120 |
result = dataset.map(predict, batched=True, batch_size=4)
|
|
|
|
| 78 |
## Evaluation
|
| 79 |
For the evaluation use the code below:
|
| 80 |
```python
|
|
|
|
|
|
|
| 81 |
import torch
|
| 82 |
import torchaudio
|
| 83 |
import librosa
|
|
|
|
| 112 |
batch["prediction"] = processor.batch_decode(logits.numpy()).text
|
| 113 |
return batch
|
| 114 |
|
| 115 |
+
dataset = load_dataset("csv", data_files={"test":"path/to/your.csv"}, delimiter=",")["test"]
|
| 116 |
+
```
|
| 117 |
+
| path| reference|
|
| 118 |
+
|---|---|
|
| 119 |
+
| path to audio files | corresponding transcription|
|
| 120 |
+
```
|
| 121 |
dataset = dataset.map(speech_file_to_array_fn)
|
| 122 |
|
| 123 |
result = dataset.map(predict, batched=True, batch_size=4)
|