Update app.py

app.py

@@ -55,6 +55,7 @@ def Generate_Chords(input_midi, input_num_prime_chords, input_num_gen_chords, in
     model = AutoregressiveWrapper(model, ignore_index = PAD_IDX, pad_value=PAD_IDX)
 
     model.to(DEVICE)
+    
     print('Done!')
     print('=' * 70)
 
@@ -63,10 +64,12 @@ def Generate_Chords(input_midi, input_num_prime_chords, input_num_gen_chords, in
     model.load_state_dict(
         torch.load('Ultimate_Chords_Progressions_Transformer_Trained_Model_LAX_5858_steps_0.4506_loss_0.8724_acc.pth',
                    map_location=DEVICE))
+
+    model.eval()
+    
     print('Done!')
     print('=' * 70)
 
-    model.eval()
 
     if DEVICE == 'cpu':
         dtype = torch.bfloat16