Update model.py

model.py

@@ -22,7 +22,7 @@ def get_model_and_tokenizer() -> Tuple[AutoModelForCausalLM, AutoTokenizer]:
         # Load the tokenizer and the model
         tokenizer = AutoTokenizer.from_pretrained("MikeMpapa/lmd_mmm_tokenizer_tutorial_artist")
         model = AutoModelForCausalLM.from_pretrained(
-            "MikeMpapa/4_bar_lmd_clean_custom_test3"
+            "MikeMpapa/4_bar_lmd_clean_custom_epochs10"
         )
 
         # Move model to device