Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ print('=' * 70)
|
|
57 |
#==================================================================================
|
58 |
|
59 |
MODEL_CHECKPOINTS = {
|
60 |
-
'without velocity - 3 epochs': '
|
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='
|
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 |
|