fix output BUG 
新版本使用此代码只会输出最后一个字符 例如2002 只会输出2,修改后可以完整输出
This commit is contained in:
QingHao 2022-08-04 00:18:40 +09:00 коммит произвёл GitHub
Родитель 0cb0758570
Коммит cb427e9ddd
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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