Upload KotobaWhisperPipeline
Browse files- kotoba_whisper.py +1 -1
kotoba_whisper.py
CHANGED
|
@@ -20,7 +20,7 @@ class Punctuator:
|
|
| 20 |
|
| 21 |
ja_punctuations = ["!", "?", "、", "。"]
|
| 22 |
|
| 23 |
-
def __init__(self, model: str = "
|
| 24 |
self.punctuation_model = PunctCapSegModelONNX.from_pretrained(model)
|
| 25 |
|
| 26 |
def punctuate(self, text: str) -> str:
|
|
|
|
| 20 |
|
| 21 |
ja_punctuations = ["!", "?", "、", "。"]
|
| 22 |
|
| 23 |
+
def __init__(self, model: str = "1-800-BAD-CODE/xlm-roberta_punctuation_fullstop_truecase"):
|
| 24 |
self.punctuation_model = PunctCapSegModelONNX.from_pretrained(model)
|
| 25 |
|
| 26 |
def punctuate(self, text: str) -> str:
|