Update handler.py
Browse files- handler.py +3 -2
handler.py
CHANGED
@@ -21,8 +21,9 @@ class EndpointHandler:
|
|
21 |
|
22 |
with torch.device("cuda"):
|
23 |
self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
|
24 |
-
|
25 |
-
|
|
|
26 |
|
27 |
embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
|
28 |
self.embeddings_dataset = embeddings_dataset
|
|
|
21 |
|
22 |
with torch.device("cuda"):
|
23 |
self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
|
24 |
+
|
25 |
+
self.processor = SpeechT5Processor.from_pretrained(checkpoint)
|
26 |
+
self.vocoder = SpeechT5HifiGan.from_pretrained(vocoder_id)
|
27 |
|
28 |
embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
|
29 |
self.embeddings_dataset = embeddings_dataset
|