Update app.py

app.py

@@ -1,7 +1,58 @@
 import gradio as gr
+import torch
+from transformers import MusicgenForConditionalGeneration, AutoProcessor
+import numpy as np
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load the model - 'small' is fastest and most reliable for free tiers
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
+processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def generate_music(prompt, duration):
+    if not prompt:
+        return None
+    
+    # Prepare the input
+    inputs = processor(text=[prompt], padding=True, return_tensors="pt")
+    
+    # Generate - 1 second is roughly 50 tokens
+    max_tokens = int(duration * 50)
+    audio_values = model.generate(**inputs, max_new_tokens=max_tokens)
+    
+    # Get sampling rate and clean up audio for Gradio
+    sampling_rate = model.config.audio_encoder.sampling_rate
+    audio_data = audio_values[0, 0].cpu().numpy()
+    
+    return (sampling_rate, audio_data)
+
+# Create a professional interface with Blocks
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""# 🎵 AI Music Studio
+    *Describe a vibe, and our neural engine will compose a unique track for your blog.*""")
+    
+    with gr.Row():
+        with gr.Column():
+            text_input = gr.Textbox(
+                label="Music Description", 
+                placeholder="e.g., 'Lofi hip hop with a relaxing rain background' or '80s synthwave'"
+            )
+            duration_slider = gr.Slider(
+                minimum=2, maximum=15, value=8, step=1, 
+                label="Duration (Seconds)"
+            )
+            generate_btn = gr.Button("Compose Music", variant="primary")
+            
+        with gr.Column():
+            audio_output = gr.Audio(label="Generated Track", type="numpy")
+            
+    generate_btn.click(
+        fn=generate_music, 
+        inputs=[text_input, duration_slider], 
+        outputs=audio_output
+    )
+
+    gr.Examples(
+        examples=[["Upbeat jazz with a fast trumpet solo", 10], ["Ambient space music with deep bass", 12]],
+        inputs=[text_input, duration_slider]
+    )
+
+demo.launch()