Change to newer model

main.py

@@ -60,7 +60,7 @@ def run():
         with gr.Row():
             with gr.Column():
                 temp = gr.Slider(
-                    minimum=0, maximum=1, step=0.05, value=0.75, label="Temperature"
+                    minimum=0, maximum=1, step=0.05, value=0.85, label="Temperature"
                 )
                 genre = gr.Dropdown(
                     choices=genres, value="POP", label="Select the genre"

model.py

@@ -22,7 +22,7 @@ def get_model_and_tokenizer() -> Tuple[AutoModelForCausalLM, AutoTokenizer]:
         # Load the tokenizer and the model
         tokenizer = AutoTokenizer.from_pretrained("juancopi81/lmd_8bars_tokenizer")
         model = AutoModelForCausalLM.from_pretrained(
-            "juancopi81/lmd-8bars-2048-epochs20_v3"
+            "juancopi81/lmd-8bars-2048-epochs40_v4"
         )
 
         # Move model to device