Update app.py

app.py

@@ -102,6 +102,10 @@ def main():
             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]