zerogpu

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 from audio2hero import generate_midi
+import spaces
+
+
 
 gradio_app = gr.Interface(
-    generate_midi,
+    spaces.GPU(generate_midi),
     inputs=gr.Audio(label="Input Audio", type="filepath"),
     outputs=gr.File(label="Output MIDI Zip File"),
     title="Audio2Hero AI Autocharter for CloneHero",

audio2hero.py

@@ -12,6 +12,7 @@ def generate_midi(song_path, output_dir=None):
 
   print("Loading Model...")
   device = "cuda" if torch.cuda.is_available() else "cpu"
+  print(f"Using {device}")
   model = Pop2PianoForConditionalGeneration.from_pretrained("Tim-gubski/Audio2Hero").to(device)
   model.eval()
   processor = Pop2PianoProcessor.from_pretrained("sweetcocoa/pop2piano")

requirements.txt

@@ -5,4 +5,5 @@ essentia
 librosa
 scipy
 pydub
-resampy
+resampy
+spaces