Update README.md

README.md

@@ -49,17 +49,17 @@ You will need the `miditok` (>=v2.1.7), `transformers` and `torch` packages to m
 import torch
 from transformers import AutoModelForCausalLM
 from miditok import TSD
-from miditoolkit import MidiFile
+from symusic import Score
 
 torch.set_default_device("cuda")
 model = AutoModelForCausalLM.from_pretrained("Natooz/Maestro-TSD-bpe20k", trust_remote_code=True, torch_dtype="auto")
 tokenizer = TSD.from_pretrained("Natooz/Maestro-TSD-bpe20k")
-input_midi = MidiFile("path/to/file.mid")
+input_midi = Score("path/to/file.mid")
 input_tokens = tokenizer(input_midi)
 
-generated_token_ids = model.generate(input_tokens.ids, max_length=200)
+generated_token_ids = model.generate(input_tokens.ids, max_length=500)
 generated_midi = tokenizer(generated_token_ids)
-generated_midi.dump("path/to/continued.mid")
+generated_midi.dump_midi("path/to/continued.mid")
 ```
 
 ## Training Details