spaces/music-gen-demo/app.py

# Install dependencies: pip install -r requirements.txt

import gradio as gr
from transformers import T5Tokenizer, T5ForConditionalGeneration

model = T5ForConditionalGeneration.from_pretrained('t5-base')
tokenizer = T5Tokenizer.from_pretrained('t5-base')

def generate_music(prompt):
    input_ids = tokenizer.encode(prompt, return_tensors='pt')
    output = model.generate(input_ids)
    return tokenizer.decode(output[0], skip_special_tokens=True)

demo = gr.Interface(
    fn=generate_music,
    inputs=gr.Textbox(label="Enter a prompt"),
    outputs=gr.Textbox(label="Generated Music")
)