change generate

generate.py

@@ -19,7 +19,7 @@ from tqdm import tqdm
 set_seed(777)
 
 # Creates the tokenizer
-tokenizer = AutoTokenizer.from_pretrained("tokenizer")
+tokenizer = REMI.from_pretrained("sunsetsobserver/MIDI")
 
 # Trains the tokenizer with Byte Pair Encoding (BPE) to build the vocabulary, here 10k tokens
 midi_paths = list(Path('Maestro').glob('**/*.mid')) + list(Path('Maestro').glob('**/*.midi'))
@@ -39,7 +39,7 @@ collator = DataCollator(
 )
 
 # Creates model using the correct configuration
-model = AutoModel.from_pretrained("./runs/model.safetensors")
+model = AutoModelForCausalLM.from_pretrained("sunsetsobserver/MIDI/runs")
 
 collator = DataCollator(tokenizer["PAD_None"], tokenizer["BOS_None"], tokenizer["EOS_None"], copy_inputs_as_labels=True)