From 5e609c1ec9976ae1536900492d702e6c1d2a28f6 Mon Sep 17 00:00:00 2001
From: tink2123 <y_tink@163.com>
Date: Sat, 18 Dec 2021 07:41:42 +0000
Subject: [PATCH] fix srn

---
 tools/program.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/program.py b/tools/program.py
index c1547efb..93c61324 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)
-- 
GitLab