Update app.py

app.py

@@ -314,7 +314,7 @@ def tokens_to_MIDI(tokens, MIDI_name):
 
 # =================================================================================================
                        
-@spaces.GPU
+# @spaces.GPU
 def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, input_sampling_overlap):
     
     print('=' * 70)
@@ -325,7 +325,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
 
     SEQ_LEN = 8192 # Models seq len
     PAD_IDX = 19463 # Models pad index
-    DEVICE = 'cuda' # 'cuda'
+    DEVICE = 'cpu' # 'cuda'
 
     # instantiate the model
 
@@ -411,7 +411,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
     
       inp = [input_src_tokens[i:i+sampling_resolution]]
     
-      inp = torch.LongTensor(inp).cuda()
+      inp = torch.LongTensor(inp).to(DEVICE)
     
       with ctx:
         with torch.no_grad():
@@ -422,7 +422,7 @@ def CompareMIDIs(input_src_midi, input_trg_midi, input_sampling_resolution, inpu
     
       inp = [input_trg_tokens[i:i+sampling_resolution]]
     
-      inp = torch.LongTensor(inp).cuda()
+      inp = torch.LongTensor(inp).to(DEVICE)
     
       with ctx:
         with torch.no_grad():