Update app.py

app.py

@@ -1,20 +1,20 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the text-to-audio model
-tts = pipeline("text-to-audio", model="facebook/musicgen-small")
+# Load the MusicGen model
+musicgen = pipeline("text-to-audio", model="facebook/musicgen-small")
 
-def text_to_audio(text):
-    output = tts(text)
+def text_to_music(prompt):
+    output = musicgen(prompt)
     return output["audio"]
 
 # Create Gradio interface
 iface = gr.Interface(
-    fn=text_to_audio,
-    inputs=gr.Textbox(label="Enter Text"),
-    outputs=gr.Audio(label="Generated Audio"),
-    title="Text-to-Audio Chatbot",
-    description="Enter text, and get an AI-generated audio response."
+    fn=text_to_music,
+    inputs=gr.Textbox(label="Enter a music description"),
+    outputs=gr.Audio(label="Generated Music"),
+    title="AI Music Generator",
+    description="Enter a description (e.g., 'happy jazz melody') and generate AI music."
 )
 
 if __name__ == "__main__":