Update app.py

app.py

@@ -163,7 +163,7 @@ def ClassifyMIDI(input_midi, input_sampling_resolution):
     
     SEQ_LEN = 1026
     PAD_IDX = 940
-    DEVICE = 'cuda' # 'cuda'
+    DEVICE = 'cpu' # 'cuda'
     
     # instantiate the model
     
@@ -216,7 +216,7 @@ def ClassifyMIDI(input_midi, input_sampling_resolution):
     
     for input in input_data:
     
-        x = torch.tensor(input[:1022], dtype=torch.long, device='cuda')
+        x = torch.tensor(input[:1022], dtype=torch.long, device=DEVICE)
     
         with ctx:
           out = model.module.generate(x,