Fixing fleurs eval
Browse files
eval.py
CHANGED
|
@@ -152,8 +152,8 @@ def main(args):
|
|
| 152 |
batch["audio"]["array"], chunk_length_s=args.chunk_length_s, stride_length_s=args.stride_length_s
|
| 153 |
)
|
| 154 |
|
| 155 |
-
batch["prediction"] = prediction[
|
| 156 |
-
batch["target"] = normalize_text(args.text_column, args.dataset)
|
| 157 |
return batch
|
| 158 |
|
| 159 |
# run inference on all examples
|
|
@@ -203,4 +203,4 @@ if __name__ == "__main__":
|
|
| 203 |
)
|
| 204 |
args = parser.parse_args()
|
| 205 |
|
| 206 |
-
main(args)
|
|
|
|
| 152 |
batch["audio"]["array"], chunk_length_s=args.chunk_length_s, stride_length_s=args.stride_length_s
|
| 153 |
)
|
| 154 |
|
| 155 |
+
batch["prediction"] = prediction["text"]
|
| 156 |
+
batch["target"] = normalize_text(batch[args.text_column], args.dataset)
|
| 157 |
return batch
|
| 158 |
|
| 159 |
# run inference on all examples
|
|
|
|
| 203 |
)
|
| 204 |
args = parser.parse_args()
|
| 205 |
|
| 206 |
+
main(args)
|