Update app.py

app.py

@@ -25,13 +25,12 @@ def generate_music_tensors(descriptions, duration: int, device):
         duration=duration
     )
 
-    # Set the device during the forward pass
-    with torch.no_grad():
+    with st.spinner("Generating Music..."):
         output = model.generate(
             descriptions=descriptions,
             progress=True,
             return_tokens=True,
-            device=device
+            device=device  # Pass the device to the generate method
         )
 
     st.success("Music Generation Complete!")
@@ -98,15 +97,12 @@ def main():
             st.subheader("Generated Music")
 
             # Generate audio
-            # descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(5)]
             descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(1)]  # Change the batch size to 1
-            music_tensors = generate_music_tensors(descriptions, time_slider)
 
             # Pass the device parameter when calling generate_music_tensors
             device = torch.device('cpu')
             music_tensors = generate_music_tensors(descriptions, time_slider, device)
 
-             # Only play the full audio for index 0
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)