get 1st row of dataset

app.py

@@ -135,7 +135,7 @@ def predict(text, speaker):
     input_ids = input_ids[..., :model.config.max_text_positions]
 
     ### ### ### 
-    example = dataset["test"][11]
+    example = dataset.iloc[0]
     speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
 
     speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)