Update app.py

app.py

@@ -13,13 +13,13 @@ def load_model():
     model = MusicGen.get_pretrained('facebook/musicgen-small')
     return model
 
-def generate_music_tensors(descriptions, duration_seconds: int):
+def generate_music_tensors(descriptions, duration: int):
     model = load_model()
 
     model.set_generation_params(
         use_sampling=True,
-        top_k=256,
-        duration=duration_seconds
+        top_k=250,
+        duration=duration
     )
 
     with st.spinner("Generating Music..."):
@@ -35,7 +35,7 @@ def generate_music_tensors(descriptions, duration_seconds: int):
 
 def save_audio(samples: torch.Tensor):
     sample_rate = 30000
-    save_path = "audio_output"
+    save_path = "/audio_output"
     assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
@@ -70,12 +70,8 @@ def main():
         # Dropdown for genres
         selected_genre = st.selectbox("Select Genre", genres)
         
-        st.subheader("2. Select time duration (In Minutes)")
-        # Adjust the time slider for selecting duration in minutes
-        time_slider = st.slider("Select time duration (In Minutes)", 1, 5, 1)
-
-        # Convert the selected minutes to seconds
-        duration_seconds = time_slider * 60
+        st.subheader("2. Select time duration (In Seconds)")
+        time_slider = st.slider("Select time duration (In Seconds)", 0, 60, 10)
 
     st.title("""🎵 Song Lab AI 🎵""")
     st.text('')
@@ -95,13 +91,13 @@ def main():
 
             # Generate audio
             descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(5)]  # Adjust the batch size (5 in this case)
-            music_tensors = generate_music_tensors(descriptions, duration_seconds)
+            music_tensors = generate_music_tensors(descriptions, time_slider)
 
-            # Only play the full audio for index 0
+             # Only play the full audio for index 0
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)
-            audio_filepath = f'audio_output/audio_{idx}.wav'
+            audio_filepath = f'/audio_output/audio_{idx}.wav'
             audio_file = open(audio_filepath, 'rb')
             audio_bytes = audio_file.read()
 
@@ -112,3 +108,4 @@ def main():
 
 if __name__ == "__main__":
     main()
+