Update app.py

app.py

@@ -242,7 +242,7 @@ def GenerateAccompaniment(input_midi, input_num_tokens, input_acc_type):
               
             ctime = cur_time
     
-            out = torch.LongTensor([[o]]).cuda()
+            out = torch.LongTensor([[o]]).to(DEVICE)
             x = torch.cat((x, out), 1)
           else:
             break