stefan-it
/

wtp-canine-s-de-hist-12l

Text2Text Generation

Model card Files Files and versions Community

stefan-it commited on Dec 3, 2023

Commit

c9aa0de

•

1 Parent(s): 72b0c64

pipeline: fix style

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -6,4 +6,4 @@ class PreTrainedPipeline:
         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="ud"))}]

         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="digibok"))}]