Update app.py

app.py

@@ -33,7 +33,7 @@ def GenerateGroove():
 
     SEQ_LEN = 4096 # Models seq len
     PAD_IDX = 1664 # Models pad index
-    DEVICE = 'cuda' # 'cuda'
+    DEVICE = 'cpu' # 'cuda'
 
     # instantiate the model
 
@@ -124,7 +124,7 @@ def GenerateGroove():
         
         while batch_value > 255 and nc < max_notes_per_chord:
         
-          x = torch.tensor([seq] * num_samples, dtype=torch.long, device='cuda')
+          x = torch.tensor([seq] * num_samples, dtype=torch.long, device=DEVICE)
         
           with ctx:
             out = model.generate(x,