Update app.py

app.py

@@ -56,21 +56,13 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINTS = {
-                    'with velocity - 3 epochs': 'Monster_Piano_Transformer_Velocity_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth',
-                    'without velocity - 3 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_69412_steps_0.8577_loss_0.7442_acc.pth',
-                    'without velocity - 7 epochs': 'Monster_Piano_Transformer_No_Velocity_Trained_Model_161960_steps_0.7775_loss_0.7661_acc.pth'
-                    }
+MODEL_CHECKPOINTS = 'Guided_Accompaniment_Transformer_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth'
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
-NUM_OUT_BATCHES = 12
-
-PREVIEW_LENGTH = 120 # in tokens
-
 #==================================================================================
 
-def load_model(model_selector):
+def load_model():
 
     print('=' * 70)
     print('Instantiating model...')
@@ -206,14 +198,10 @@ def save_midi(tokens, batch_number=None, model_selector=''):
 #==================================================================================
 
 @spaces.GPU
-def generate_music(prime, 
-                   num_gen_tokens,
-                   num_mem_tokens,
-                   num_gen_batches,
-                   model_temperature,
-                   # model_sampling_top_p,
-                   model_state
-                  ):
+def generate_accompaniment(input_midi, 
+                           num_gen_tokens,
+                           model_temperature
+                          ):
 
     if not prime:
         inputs = [0]
@@ -391,26 +379,6 @@ def generate_callback_wrapper(input_midi,
 
 #==================================================================================
 
-def reset(final_composition=[], generated_batches=[], block_lines=[], model_state=[]):
-    
-    final_composition = []
-    generated_batches = []
-    block_lines = []
-    model_state = []
-
-    return final_composition, generated_batches, block_lines
-    
-#==================================================================================
-
-def reset_demo(final_composition=[], generated_batches=[], block_lines=[], model_state=[]):
-    
-    final_composition = []
-    generated_batches = []
-    block_lines = []
-    model_state = []
-
-#==================================================================================
-
 PDT = timezone('US/Pacific')
 
 print('=' * 70)
@@ -470,7 +438,7 @@ with gr.Blocks() as demo:
     generate_btn.click(generate_accompaniment, 
                        [input_midi, 
                         num_gen_tokens,
-                        model_temperature,
+                        model_temperature
                        ], 
                        [
                         output_audio,