Method post_process

utils/gemini_inference.py:42–47

(self, text)

        )

    def post_process(self, text):
        match = re.search(r"(?i)(?<=\banswer:\s).*", text)
        if match:
            return match.group(0)
        else:
            return text

    def predict(self, prompt):
        chat_session = self.model.start_chat(