Update app.py

app.py

@@ -202,7 +202,9 @@ def Generate_Chords_Textures(input_midi,
     print('=' * 70)
     print('Generating...')
 
-    x = torch.LongTensor([705] + chords[:128] + [706]).cuda()
+    seq = [705] + chords[:128] + [706] + score[:3]
+
+    x = torch.LongTensor().cuda(seq)
     
     with ctx:
         out = model.generate(x,
@@ -210,11 +212,11 @@ def Generate_Chords_Textures(input_midi,
                              temperature=model_temperature,
                              filter_logits_fn=top_p,
                              filter_kwargs={'thres': model_sampling_top_p},
-                             return_prime=False,
+                             return_prime=True,
                              eos_token=707,
                              verbose=False)
     
-    final_song = out.tolist()
+    final_song = out.tolist()[len(seq)-3:]
 
     #==================================================================