Update app.py

app.py

@@ -17,7 +17,9 @@ def load_model():
     return model
 
 def generate_music_tensors(descriptions, duration: int):
-    model = load_model()
+    # model = load_model()
+    model = load_model().to('cpu')
+
 
     model.set_generation_params(
         use_sampling=True,
@@ -96,7 +98,8 @@ def main():
             st.subheader("Generated Music")
 
             # Generate audio
-            descriptions = [f"{text_area} {selected_genre} {bpm} BPM" for _ in range(5)]  # Adjust the batch size (5 in this case)
+            # 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)
 
              # Only play the full audio for index 0