Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
@@ -12,6 +12,7 @@ class EndpointHandler:
|
|
12 |
self.embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
|
13 |
|
14 |
def __call__(self, data):
|
|
|
15 |
text = data.get("inputs", "")
|
16 |
# Extract speaker_embedding using the index from your dataset, or replace with your own logic.
|
17 |
speaker_embedding = torch.tensor(self.embeddings_dataset[7306]["xvector"]).unsqueeze(0)
|
|
|
12 |
self.embeddings_dataset = load_dataset("Matthijs/cmu-arctic-xvectors", split="validation")
|
13 |
|
14 |
def __call__(self, data):
|
15 |
+
|
16 |
text = data.get("inputs", "")
|
17 |
# Extract speaker_embedding using the index from your dataset, or replace with your own logic.
|
18 |
speaker_embedding = torch.tensor(self.embeddings_dataset[7306]["xvector"]).unsqueeze(0)
|