Spaces:

asigalov61
/

Allegro-Music-Transformer

Running on Zero

asigalov61 commited on Sep 15, 2023

Commit

bbf9a0d

•

1 Parent(s): 498b808

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -182,7 +182,7 @@ if __name__ == "__main__":
     print('Loading model checkpoint...')
-    model.load_state_dict(torch.load(full_path_to_model_checkpoint, map_location='cpu'))
     print('=' * 70)
     model.eval()

     print('Loading model checkpoint...')
+    model.load_state_dict(torch.load('Allegro_Music_Transformer_Small_Trained_Model_56000_steps_0.9399_loss_0.7374_acc.pth', map_location='cpu'))
     print('=' * 70)
     model.eval()