fix answer
This commit is contained in:
parent
dcbeb65f76
commit
6dc88090fb
|
@ -46,5 +46,5 @@ class InferenceService:
|
|||
early_stopping=True,
|
||||
no_repeat_ngram_size=2
|
||||
)
|
||||
responses = [self.tokenizer.decode(out, skip_special_tokens=True).replace("Ожидаемый ответ:", "").strip() for out in outputs]
|
||||
responses = [self.tokenizer.decode(out, skip_special_tokens=True).replace("Ожидаемый ответ:", "").replace("Запрос:", "Исхоный текст:").strip() for out in outputs]
|
||||
return responses
|
||||
|
|
Loading…
Reference in New Issue