| from typing import List, Dict | |
| from wtpsplit import WtP | |
| class PreTrainedPipeline: | |
| def __init__(self, path=""): | |
| self.model = WtP(model_name_or_model="wtp-canine-s-de-hist-12l", hub_prefix="stefan-it") | |
| def __call__(self, inputs: str) -> List[Dict]: | |
| return [{"generated_text": "</s>\n".join(sent.strip() for sent in self.model.split(inputs.strip(), lang_code="de", style="onb-ner"))}] | |