Dupaja
/

speecht5_tts

Inference Endpoints

Model card Files Files and versions Community

Dupaja commited on Jan 2, 2024

Commit

9cf2190

·

1 Parent(s): bd9bb8e

Update handler.py

Files changed (1) hide show

handler.py +3 -2

handler.py CHANGED Viewed

@@ -23,8 +23,9 @@ class EndpointHandler:
             self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
             self.processor = SpeechT5Processor.from_pretrained(checkpoint)
             self.vocoder = SpeechT5HifiGan.from_pretrained(vocoder_id)
-            embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
-            self.embeddings_dataset = embeddings_dataset
         self.speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)

             self.model= SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
             self.processor = SpeechT5Processor.from_pretrained(checkpoint)
             self.vocoder = SpeechT5HifiGan.from_pretrained(vocoder_id)
+        embeddings_dataset = load_dataset(dataset_id, split="validation", trust_remote_code=True)
+        self.embeddings_dataset = embeddings_dataset
         self.speaker_embeddings = torch.tensor(embeddings_dataset[7306]["xvector"]).unsqueeze(0)