Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
@@ -20,7 +20,7 @@ else:
|
|
20 |
class EndpointHandler():
|
21 |
def __init__(self, path=""):
|
22 |
# Load all required models
|
23 |
-
self.model_id = "Oysiyl/
|
24 |
self.spk_model_name = "speechbrain/spkrec-xvect-voxceleb"
|
25 |
self.model = SpeechT5ForTextToSpeech.from_pretrained(self.model_id, torch_dtype=dtype).to(device)
|
26 |
self.processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
|
@@ -81,7 +81,7 @@ class EndpointHandler():
|
|
81 |
('і', 'i'),
|
82 |
('ї', 'ji'),
|
83 |
('ґ', 'g')
|
84 |
-
|
85 |
|
86 |
for src, dst in replacements:
|
87 |
text = text.replace(src, dst)
|
|
|
20 |
class EndpointHandler():
|
21 |
def __init__(self, path=""):
|
22 |
# Load all required models
|
23 |
+
self.model_id = "Oysiyl/speecht5_tts_common_voice_nl"
|
24 |
self.spk_model_name = "speechbrain/spkrec-xvect-voxceleb"
|
25 |
self.model = SpeechT5ForTextToSpeech.from_pretrained(self.model_id, torch_dtype=dtype).to(device)
|
26 |
self.processor = SpeechT5Processor.from_pretrained("microsoft/speecht5_tts")
|
|
|
81 |
('і', 'i'),
|
82 |
('ї', 'ji'),
|
83 |
('ґ', 'g')
|
84 |
+
]
|
85 |
|
86 |
for src, dst in replacements:
|
87 |
text = text.replace(src, dst)
|