Update app.py

app.py

@@ -59,7 +59,7 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINT = 'Godzilla_Piano_Chords_Texturing_Trained_Model_36457_steps_0.5384_loss_0.8417_acc.pth'
+MODEL_CHECKPOINT = 'Godzilla_Piano_Chords_Texturing_Trained_Model_18001_steps_0.8099_loss_0.7677_acc.pth'
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
@@ -69,21 +69,6 @@ MAX_GEN_TOKS = 3072
 
 #==================================================================================
 
-print('=' * 70)
-print('Loading popular hook melodies dataset...')
-
-popular_hook_melodies_pickle = hf_hub_download(repo_id='projectlosangeles/Godzilla-Piano-Chords-Texturing', 
-                                               filename='popular_hook_melodies_24_64_CC_BY_NC_SA.pickle'
-                                              )
-
-popular_hook_melodies = TMIDIX.Tegridy_Any_Pickle_File_Reader(popular_hook_melodies_pickle)
-
-print('=' * 70)
-print('Done!')
-print('=' * 70)
-
-#==================================================================================
-
 print('=' * 70)
 print('Instantiating model...')
 
@@ -112,9 +97,9 @@ model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 print('=' * 70)
 print('Loading model checkpoint...')      
 
-model_checkpoint = hf_hub_download(repo_id='projectlosangeles/Godzilla-Piano-Chords-Texturing', filename=MODEL_CHECKPOINT)
+model_checkpoint = hf_hub_download(repo_id='asigalov61/Godzilla-Piano-Transformer', filename=MODEL_CHECKPOINT)
 
-model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))
+model.load_state_dict(torch.load(model_checkpoint, map_location=device_type, weights_only=True))
 
 model = torch.compile(model, mode='max-autotune')
 
@@ -425,11 +410,6 @@ with gr.Blocks() as demo:
     input_midi = gr.File(label="Input MIDI", 
                          file_types=[".midi", ".mid", ".kar"]
                         )
-
-    input_melody = gr.Textbox(value="Hotel California", 
-                              label="Popular melodies database search query",
-                              info='If the query is not found, random melody will be selected. Custom MIDI overrides search query'
-                             )
     
     gr.Markdown("## Generation options")
     
@@ -449,7 +429,6 @@ with gr.Blocks() as demo:
 
     generate_btn.click(Generate_Accompaniment, 
                        [input_midi,
-                        input_melody,
                         melody_patch,
                         use_nth_note,
                         model_temperature,
@@ -463,10 +442,9 @@ with gr.Blocks() as demo:
                       )
 
     gr.Examples(
-                [["Sharing The Night Together.kar", "Custom MIDI", -1, 1, 0.9, 15]
+                [["Sharing The Night Together.kar", -1, 1, 0.9, 15]
                 ],
                 [input_midi,
-                 input_melody,
                  melody_patch,
                  use_nth_note,
                  model_temperature,