diff --git a/tools/program.py b/tools/program.py index c1547efbcebc5ee8522aa7f190c44d602b595880..93c61324285f0e6128727578c18627ee4be43791 100755 --- a/tools/program.py +++ b/tools/program.py @@ -403,7 +403,7 @@ def eval(model, start = time.time() if model_type == 'table' or extra_input: preds = model(images, data=batch[1:]) - if model_type == "kie": + elif model_type == "kie": preds = model(batch) else: preds = model(images)