diff --git a/summarize_service/app/inference_service.py b/summarize_service/app/inference_service.py index 37061f6..559cc10 100644 --- a/summarize_service/app/inference_service.py +++ b/summarize_service/app/inference_service.py @@ -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