Update app.py

app.py

@@ -28,7 +28,7 @@ from midi_to_colab_audio import midi_to_colab_audio
 
 # =================================================================================================
 
-@spaces.GPU
+# @spaces.GPU
 def Harmonize_Melody(input_src_midi,
                     source_melody_transpose_value,
                     model_top_k_sampling_value,
@@ -117,13 +117,13 @@ def Harmonize_Melody(input_src_midi,
     
     model_path = 'Melody_Harmonizer_Transformer_Trained_Model_14961_steps_0.4155_loss_0.8664_acc.pth'
     
-    model.load_state_dict(torch.load(model_path))
+    model.load_state_dict(torch.load(model_path, map_location='cpu'))
     
-    model.cuda()
+    model.cpu()
     
     dtype = torch.bfloat16
     
-    ctx = torch.amp.autocast(device_type='cuda', dtype=dtype)
+    ctx = torch.amp.autocast(device_type='cpu', dtype=dtype)
     
     model.eval()
     
@@ -151,7 +151,7 @@ def Harmonize_Melody(input_src_midi,
     
         data.append(mel_chunk[j])
     
-        x = torch.tensor([data], dtype=torch.long, device='cuda')
+        x = torch.tensor([data], dtype=torch.long, device='cpu')
     
         with ctx:
           out = model.generate(x,