fix output BUG
fix output BUG 新版本使用此代码只会输出最后一个字符 例如2002 只会输出2,修改后可以完整输出
This commit is contained in:
Родитель
0cb0758570
Коммит
cb427e9ddd
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче