added spaces.gpu

app.py

@@ -31,7 +31,7 @@ def save_wav(filepath):
     return wav_filepath
 
 
-def generate_midi(caption, temperature=0.9, max_len=3000):
+def generate_midi(caption, temperature=0.9, max_len=1000):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     artifact_folder = 'artifacts'
 
@@ -60,10 +60,10 @@ def generate_midi(caption, temperature=0.9, max_len=3000):
     generated_midi = r_tokenizer.decode(output_list)
     generated_midi.dump_midi("output.mid")
 
-# @spaces.GPU(duration=120)
-def gradio_generate(prompt, temperature):
+@spaces.GPU
+def gradio_generate(prompt, temperature, max_length):
     # Generate midi
-    generate_midi(prompt, temperature)
+    generate_midi(prompt, temperature, max_length)
 
     # Convert midi to wav
     filename = "output.mid"
@@ -88,6 +88,7 @@ Generate midi music using Text2midi by providing a text prompt.
 input_text = gr.Textbox(lines=2, label="Prompt")
 output_audio = gr.Audio(label="Generated Music", type="filepath")
 temperature = gr.Slider(minimum=0.5, maximum=1.2, value=1.0, step=0.1, label="Temperature", interactive=True)
+max_length = gr.Number(default=1000, label="Max Length", min=100, max=2000, step=100)
 
 # CSS styling for the Duplicate button
 css = '''
@@ -102,7 +103,7 @@ css = '''
 # Gradio interface
 gr_interface = gr.Interface(
     fn=gradio_generate,
-    inputs=[input_text, temperature],
+    inputs=[input_text, temperature, max_length],
     outputs=[output_audio],
     description=description_text,
     allow_flagging=False,