fix answer
This commit is contained in:
parent
dcbeb65f76
commit
6dc88090fb
|
@ -46,5 +46,5 @@ class InferenceService:
|
||||||
early_stopping=True,
|
early_stopping=True,
|
||||||
no_repeat_ngram_size=2
|
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
|
return responses
|
||||||
|
|
Loading…
Reference in New Issue