Update app.py

app.py

@@ -1,18 +1,28 @@
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
+from IPython.display import Audio
 
-
+# Load the processor and model
 processor = AutoProcessor.from_pretrained("facebook/musicgen-large")
 model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-large")
 
-inputs = processor(
-    text=["80s pop track with bassy drums and synth", "90s rock song with loud guitars and heavy drums"],
-    padding=True,
-    return_tensors="pt",
-)
+# Function to generate music from text
+def generate_music_from_text(text):
+    inputs = processor(
+        text=[text],
+        padding=True,
+        return_tensors="pt",
+    )
 
-audio_values = model.generate(**inputs, max_new_tokens=256)
+    audio_values = model.generate(**inputs, max_new_tokens=256)
 
-from IPython.display import Audio
+    sampling_rate = model.config.audio_encoder.sampling_rate
+    return audio_values[0].numpy(), sampling_rate
+
+# Input text
+text_input = input("Enter the text you want to transform into music: ")
+
+# Generate music from the input text
+audio_data, sampling_rate = generate_music_from_text(text_input)
 
-sampling_rate = model.config.audio_encoder.sampling_rate
-Audio(audio_values[0].numpy(), rate=sampling_rate)
+# Display the generated music as audio
+Audio(audio_data, rate=sampling_rate)