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)
|