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