patchbanks commited on
Commit
892b802
·
verified ·
1 Parent(s): e8e6970

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -83,6 +83,7 @@ def clear_midi(dir):
83
  clear_midi(temp_dir)
84
 
85
 
 
86
  def generate_midi(temperature, top_k):
87
  start_ids = encode(start)
88
  x = (torch.tensor(start_ids, dtype=torch.long, device=device)[None, ...])
 
83
  clear_midi(temp_dir)
84
 
85
 
86
+ @spaces.GPU(duration=15)
87
  def generate_midi(temperature, top_k):
88
  start_ids = encode(start)
89
  x = (torch.tensor(start_ids, dtype=torch.long, device=device)[None, ...])