asigalov61 commited on
Commit
51677ef
·
verified ·
1 Parent(s): 3764fc5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ print('=' * 70)
57
  #==================================================================================
58
 
59
  MODEL_CHECKPOINTS = {
60
- 'without velocity - 3 epochs': 'Godzilla_Piano_Transformer_No_Velocity_Trained_Model_8038_steps_0.6459_loss_0.8154_acc.pth'
61
  }
62
 
63
  SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
@@ -98,7 +98,7 @@ print('=' * 70)
98
  print('Loading model checkpoint...')
99
 
100
  model_checkpoint = hf_hub_download(repo_id='asigalov61/Godzilla-Piano-Transformer',
101
- filename='Godzilla_Piano_Transformer_No_Velocity_Trained_Model_8038_steps_0.6459_loss_0.8154_acc.pth')
102
 
103
  model.load_state_dict(torch.load(model_checkpoint, map_location='cuda', weights_only=True))
104
 
 
57
  #==================================================================================
58
 
59
  MODEL_CHECKPOINTS = {
60
+ 'without velocity - 3 epochs': 'Godzilla_Piano_Transformer_No_Velocity_Trained_Model_14075_steps_0.4534_loss_0.8687_acc.pth'
61
  }
62
 
63
  SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
98
  print('Loading model checkpoint...')
99
 
100
  model_checkpoint = hf_hub_download(repo_id='asigalov61/Godzilla-Piano-Transformer',
101
+ filename='Godzilla_Piano_Transformer_No_Velocity_Trained_Model_14075_steps_0.4534_loss_0.8687_acc.pth')
102
 
103
  model.load_state_dict(torch.load(model_checkpoint, map_location='cuda', weights_only=True))
104