fix srn

5e609c1e · tink2123 · 5e7ad64c · 5e609c1e
Commit 5e609c1e authored Dec 18, 2021 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tools/program.py tools/program.py +1 -1

No files found.
--- 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)