Update app.py

app.py

@@ -36,7 +36,7 @@ PDT = timezone('US/Pacific')
 
 MODEL_CHECKPOINT = 'Orpheus_Music_Transformer_Trained_Model_91168_steps_0.63_loss_0.8352_acc.pth'
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
-NUM_OUT_BATCHES = 12
+NUM_OUT_BATCHES = 8
 PREVIEW_LENGTH = 120  # in tokens
 
 # -----------------------------
@@ -272,9 +272,9 @@ def save_midi(tokens):
 # MUSIC GENERATION FUNCTION (Combined)
 # -----------------------------
 @spaces.GPU
-def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model_temperature, model_top_p):
+def generate_music(prime, num_gen_tokens, num_gen_batches, model_temperature, model_top_p):
     """Generate music tokens given prime tokens and parameters."""
-    inputs = prime[-num_mem_tokens:] if prime else [18816]
+    inputs = prime if prime else [18816]
     print("Generating...")
     inp = torch.LongTensor([inputs] * num_gen_batches).cuda()
     with ctx:
@@ -293,7 +293,7 @@ def generate_music(prime, num_gen_tokens, num_mem_tokens, num_gen_batches, model
     print_sep()
     return out.tolist()
 
-def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_mem_tokens,
+def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens,
                              model_temperature, model_top_p, add_drums, add_outro, final_composition, generated_batches, block_lines):
     """
     Generate tokens using the model, update the composition state, and prepare outputs.
@@ -310,7 +310,6 @@ def generate_music_and_state(input_midi, num_prime_tokens, num_gen_tokens, num_m
 
     print('Num prime tokens:', num_prime_tokens)
     print('Num gen tokens:', num_gen_tokens)
-    print('Num mem tokens:', num_mem_tokens)
 
     print('Model temp:', model_temperature)
     print('Model top p:', model_top_p)
@@ -414,10 +413,12 @@ def remove_batch(batch_number, num_tokens, final_composition, generated_batches,
 
 def clear():
     """Clear outputs and reset state."""
+    print('Clear batch...')
     return None, None, None, [], []
 
 def reset(final_composition=[], generated_batches=[], block_lines=[]):
     """Reset composition state."""
+    print('Reset...')
     return [], [], []
 
 # -----------------------------
@@ -462,7 +463,6 @@ with gr.Blocks() as demo:
     gr.Markdown("## Generate")
     num_prime_tokens = gr.Slider(16, 7168, value=7168, step=1, label="Number of prime tokens")
     num_gen_tokens = gr.Slider(16, 1024, value=512, step=1, label="Number of tokens to generate")
-    num_mem_tokens = gr.Slider(16, 8192, value=8192, step=1, label="Number of memory tokens")
     model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
     model_top_p = gr.Slider(0.1, 0.99, value=0.96, step=0.01, label="Model sampling top p value")
     add_drums = gr.Checkbox(value=False, label="Add drums")