Update app.py

app.py

@@ -1,6 +1,21 @@
 import gradio as gr
+from transformers import pipeline
 
-gr.load(
-   "models/facebook/musicgen-small",
-   provider="hf-inference",
-).launch()
+# Load the text-to-audio model
+tts = pipeline("text-to-audio", model="facebook/musicgen-small")
+
+def text_to_audio(text):
+    output = tts(text)
+    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."
+)
+
+if __name__ == "__main__":
+    iface.launch()