From dcbeb65f76fa4efd38996598ec03b9d9156dfc66 Mon Sep 17 00:00:00 2001 From: itqop Date: Mon, 3 Mar 2025 04:56:03 +0300 Subject: [PATCH] fix anwser --- summarize_service/app/inference_service.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/summarize_service/app/inference_service.py b/summarize_service/app/inference_service.py index 4da6d27..37061f6 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) for out in outputs] + responses = [self.tokenizer.decode(out, skip_special_tokens=True).replace("Ожидаемый ответ:", "").strip() for out in outputs] return responses