Update README.md

README.md

@@ -36,15 +36,15 @@ How to use ?
 
 ```python
 from musiclang_predict import predict_chords, MusicLangTokenizer
-from transformers import GPT2LMHeadModel
+from transformers import AutoModelForCausalLM, AutoTokenizer
 from musiclang.library import *
 
 # Load model and tokenizer
-model = GPT2LMHeadModel.from_pretrained('musiclang/musiclang-chord-v2-4k')
-tokenizer = MusicLangTokenizer('musiclang/musiclang-chord-v2-4k')
-soundtrack = predict_chords(model, tokenizer, chord_duration=4, nb_chords=2)
+model = AutoModelForCausalLM.from_pretrained('musiclang/musiclang-chord-v2-4k')
+tokenizer = AutoTokenizer.from_pretrained('musiclang/musiclang-chord-v2-4k')
+soundtrack = predict_chords(model, tokenizer, nb_chords=4, temperature=1.0)
 
-# Gave the chord a simple voicing
+# Give the chord a simple voicing (closed position chord)
 soundtrack = soundtrack(b0, b1, b2, b3)
 
 # Save it to midi