Update app.py

app.py

@@ -56,7 +56,7 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINTS = 'Guided_Accompaniment_Transformer_Trained_Model_59896_steps_0.9055_loss_0.735_acc.pth'
+MODEL_CHECKPOINT = 'Guided_Accompaniment_Transformer_Trained_Model_36457_steps_0.5384_loss_0.8417_acc.pth'
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
@@ -90,7 +90,7 @@ model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 print('=' * 70)
 print('Loading model checkpoint...')      
 
-model_checkpoint = hf_hub_download(repo_id='asigalov61/Guided-Accompaniment-Transformer', filename=MODEL_CHECKPOINTS[model_selector])
+model_checkpoint = hf_hub_download(repo_id='asigalov61/Guided-Accompaniment-Transformer', filename=MODEL_CHECKPOINT)
 
 model.load_state_dict(torch.load(model_checkpoint, map_location='cpu', weights_only=True))