Update app.py

app.py

@@ -15,20 +15,18 @@ import random
 import spaces
 import os
 
-
 torch.backends.cudnn.benchmark = True
 torch.backends.cudnn.allow_tf32 = True
 torch.backends.cuda.matmul.allow_tf32 = True
 
 in_space = os.getenv("SYSTEM") == "spaces"
-os.environ['CUDA_HOME'] = '/usr/local/cuda'
 
 n_layer = 12
 n_embd = 768
 ctx_len = 2048
 
 os.environ['RWKV_FLOAT_MODE'] = 'fp32'
-os.environ['RWKV_RUN_DEVICE'] = 'cuda'
+os.environ['RWKV_RUN_DEVICE'] = 'cpu'
 model_type = 'RWKV'
 
 MODEL_NAME = 'model'
@@ -71,7 +69,6 @@ def humanize_notes(midi_events):
     return midi_events
 
 
-@spaces.GPU(duration=60)
 def generate_midi(LENGTH_PER_TRIAL, src_ctx, model, src_len, ctx_len, TEMPERATURE, top_k, tokenizer, ctx_seed, bpm):
     midi_seq = []