Spaces:
Runtime error
Runtime error
divakaivan
commited on
get 1st row of dataset
Browse files
app.py
CHANGED
@@ -135,7 +135,7 @@ def predict(text, speaker):
|
|
135 |
input_ids = input_ids[..., :model.config.max_text_positions]
|
136 |
|
137 |
### ### ###
|
138 |
-
example = dataset[
|
139 |
speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
|
140 |
|
141 |
speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
|
|
|
135 |
input_ids = input_ids[..., :model.config.max_text_positions]
|
136 |
|
137 |
### ### ###
|
138 |
+
example = dataset.iloc[0]
|
139 |
speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
|
140 |
|
141 |
speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
|