cuda support

app.py

@@ -1,3 +1,4 @@
+import torch
 import gradio as gr
 import os
 from transformer_wrapper import TransformerWrapper
@@ -18,6 +19,9 @@ def model_load():
  )
  model_id = "dpipqxiy"
  wrapper.eval()
+ if torch.cuda.is_available():
+ wrapper = wrapper.cuda()
+
  return wrapper, model_id, config
 
 
@@ -37,7 +41,7 @@ def inference(file_up, composer):
  show_plot=False,
  save_midi=True,
  save_mix=True,
- midi_path="output.mid"
+ midi_path="output.mid",
  )
 
  return mix_path, midi_path