Spaces:

asigalov61
/

Guided-Rock-Music-Transformer

Running on Zero

asigalov61 commited on Sep 25

Commit

0764984

•

1 Parent(s): 00d6bd7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ def Generate_Rock_Song(input_midi,
     print('Loading model checkpoint...')
     model.load_state_dict(
-        torch.load('Melody2Song_Seq2Seq_Music_Transformer_Trained_Model_28482_steps_0.719_loss_0.7865_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)

     print('Loading model checkpoint...')
     model.load_state_dict(
+        torch.load('Guided_Rpck_Music_Transformer_Trained_Model_12081_steps_0.4113_loss_0.8747_acc.pth',
                    map_location=DEVICE))
     print('=' * 70)