versae commited on
Commit
df9222e
1 Parent(s): bc4c4dc

Fixing fleurs eval

Browse files
Files changed (1) hide show
  1. eval.py +3 -3
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[args.text_column]
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)