Update app.py

app.py

@@ -231,7 +231,9 @@ def generate_music(
                 else:
                     raw_output = output_seq
 
-        generator()
+            return raw_output
+
+        raw_output = generator()   
         
         # save raw output and check sanity
         ids = raw_output[0].cpu().numpy()