Update app.py

app.py

@@ -54,6 +54,8 @@ print('=' * 70)
 
 #==================================================================================
 
+MODEL_CHECKPOINT = 'Giant_Music_Transformer_Medium_Trained_Model_25603_steps_0.3799_loss_0.8934_acc.pth'
+
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
 NUM_OUT_BATCHES = 8
@@ -90,9 +92,7 @@ model = AutoregressiveWrapper(model, ignore_index=PAD_IDX, pad_value=PAD_IDX)
 print('=' * 70)
 print('Loading model checkpoint...')
 
-model_path = 'Giant_Music_Transformer_Medium_Trained_Model_20355_steps_0.709_loss_0.812_acc.pth'
-
-model.load_state_dict(torch.load(model_path, map_location='cpu'))
+model.load_state_dict(torch.load(MODEL_CHECKPOINT, map_location='cpu'))
 
 print('=' * 70)
 print('Done!')