Create spaces/music-gen-demo/app.py

spaces/music-gen-demo/app.py

@@ -0,0 +1,16 @@
+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")
+)