diff --git a/examples/machine_reading_comprehension/SQuAD/deploy/python/predict.py b/examples/machine_reading_comprehension/SQuAD/deploy/python/predict.py index 665659dd8db4e64f58f702de984d3a5f00fba6c9..e68e604beeb6b2604974619067cfd7da4dc3308d 100644 --- a/examples/machine_reading_comprehension/SQuAD/deploy/python/predict.py +++ b/examples/machine_reading_comprehension/SQuAD/deploy/python/predict.py @@ -115,7 +115,7 @@ def main(): os.path.dirname(args.model_name_or_path)) if args.predict_file: - dataset = load_dataset('sqaud', data_files=args.predict_file) + dataset = load_dataset('squad', data_files=args.predict_file) elif args.version_2_with_negative: dataset = load_dataset('squad', splits='dev_v2') else: