display source

app.py

@@ -15,12 +15,12 @@ def main():
     )
     piano_task = PianoTasks.get_task(task_name=task_name)
 
-    dataset = load_dataset("epr-labs/maestro-sustain-v2", split="test")
+    dataset = load_dataset("epr-labs/maestro-sustain-v2", split="test+validation")
     record_idx = st.number_input(
         label="Record ID",
         min_value=0,
         max_value=len(dataset) - 1,
-        value=12,
+        value=123,
     )
 
     record = dataset[record_idx]
@@ -35,6 +35,7 @@ def main():
         piece=source_piece,
         secondary_piece=target_piece,
     )
+    st.json(piece.source)
 
 
 if __name__ == "__main__":