dygie_input.json error

[BodongDu]

Traceback (most recent call last):
File "/home/bduag/anaconda3/envs/cxr-old/bin/allennlp", line 8, in
sys.exit(run())
File "/home/bduag/anaconda3/envs/cxr-old/lib/python3.7/site-packages/allennlp/main.py", line 34, in run
main(prog="allennlp")
File "/home/bduag/anaconda3/envs/cxr-old/lib/python3.7/site-packages/allennlp/commands/init.py", line 92, in main
args.func(args)
File "/home/bduag/anaconda3/envs/cxr-old/lib/python3.7/site-packages/allennlp/commands/predict.py", line 226, in _predict
manager.run()
File "/home/bduag/anaconda3/envs/cxr-old/lib/python3.7/site-packages/allennlp/commands/predict.py", line 193, in run
for model_input_instance, result in zip(batch, self._predict_instances(batch)):
File "/home/bduag/anaconda3/envs/cxr-old/lib/python3.7/site-packages/allennlp/commands/predict.py", line 152, in _predict_instances
results = [self._predictor.predict_instance(batch_data[0])]
File "/home/hyangdh/MRG/dygiepp/dygie/predictors/dygie.py", line 56, in predict_instance
prediction = model.make_output_human_readable(model(**model_input)).to_json()
File "/home/hyangdh/MRG/dygiepp/dygie/models/dygie.py", line 323, in make_output_human_readable
for predictions, sentence in zip(output_dict["ner"]["predictions"], doc):

It seems that the input.json missed the dataset key, how can I fix it?