Update app.py

app.py

@@ -33,7 +33,7 @@ def GenerateSong(input_melody_seed_number):
 
     SEQ_LEN = 2560
     PAD_IDX = 514
-    DEVICE = 'cuda' # 'cuda'
+    DEVICE = 'cpu' # 'cuda'
 
     # instantiate the model
 
@@ -78,7 +78,7 @@ def GenerateSong(input_melody_seed_number):
     print('=' * 70)
     print('Generating...')
 
-    x = (torch.tensor(seed_melody, dtype=torch.long, device='cuda')[None, ...])
+    x = (torch.tensor(seed_melody, dtype=torch.long, device=DEVICE)[None, ...])
 
     with ctx:
         with torch.inference_mode():