Update app.py

app.py

@@ -56,8 +56,7 @@ print('=' * 70)
 
 #==================================================================================
 
-MODEL_CHECKPOINT = 'Orpheus_Bridge_Music_Transformer_Trained_Model_19571_steps_0.9396_loss_0.7365_acc.pth'
-# MODEL_CHECKPOINT = 'Orpheus_Bridge_Music_Transformer_Trained_Model_43450_steps_0.8334_loss_0.7629_acc.pth'
+MODEL_CHECKPOINT = 'Orpheus_Bridge_Music_Transformer_Trained_Model_43450_steps_0.8334_loss_0.7629_acc.pth'
 
 SOUDFONT_PATH = 'SGM-v2.01-YamahaGrand-Guit-Bass-v2.7.sf2'
 
@@ -255,6 +254,9 @@ def Generate_Music_Bridge(input_midi,
                     rcount += 1
                     print('Regenerating attempt #', rcount)
 
+                    if rcount == 3:
+                        break
+
             final_song = left_chunk + y[:-64] + right_chunk
         
             #==================================================================