fix anwser

2025-03-03 04:56:03 +03:00 · 2025-03-03 04:56:03 +03:00 · dcbeb65f76
parent e7f7081bc6
commit dcbeb65f76
1 changed files with 1 additions and 1 deletions
--- 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