Update app.py

app.py

@@ -59,10 +59,10 @@ def GenerateMIDI():
             break
             
     if return_prime:
-      return out[:, :]
+      melody_chords_f = out[:, :]
     
     else:
-      return out[:, st:]
+      melody_chords_f = out[:, st:]
 
     melody_chords_f = melody_chords_f.tolist()[0]