Giant-Music-Transformer

Running on Zero

asigalov61 commited on Oct 28, 2024

Commit

7b40a3e

verified ·

1 Parent(s): 6da0de9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -250,7 +250,7 @@ def generate_music(prime,
     print('=' * 70)
     print('Loading model checkpoint...')
-    model_path = 'Giant-Music-Transformer/Models/Medium/Giant_Music_Transformer_Medium_Trained_Model_10446_steps_0.7202_loss_0.8233_acc.pth'
     model.load_state_dict(torch.load(model_path))

     print('=' * 70)
     print('Loading model checkpoint...')
+    model_path = 'Giant_Music_Transformer_Medium_Trained_Model_10446_steps_0.7202_loss_0.8233_acc.pth'
     model.load_state_dict(torch.load(model_path))