From cb427e9ddd6898420388323d2789ee99d1d8c168 Mon Sep 17 00:00:00 2001
From: QingHao <462902059@qq.com>
Date: Thu, 4 Aug 2022 00:18:40 +0900
Subject: [PATCH] fix output BUG
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

fix output BUG
新版本使用此代码只会输出最后一个字符 例如2002 只会输出2，修改后可以完整输出
---
 tapex/model_interface.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tapex/model_interface.py b/tapex/model_interface.py
index 3279839..1d721b1 100644
--- a/tapex/model_interface.py
+++ b/tapex/model_interface.py
@@ -36,4 +36,4 @@ class TAPEXModelInterface:
             beam=5
         )
         # the result should be a list of answers, and we only care about the answer itself instead of score
-        return model_output[0][0]
+        return model_output