update

app.py

@@ -177,7 +177,13 @@ def generate_music(seed, use_chords, chord_progression, prompt_duration, musicge
             seed = random.randint(1, 10000)
 
 # Check if CUDA is available
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+if torch.cuda.is_available():
+  device = "cuda"
+elif torch.backends.mps.is_available():
+  device = "mps"
+else:
+  device = "cpu"
+#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Define the expandable sections
 musiclang_blurb = """
@@ -236,4 +242,4 @@ with gr.Blocks() as iface:
 
     generate_button.click(generate_music, inputs=[seed, use_chords, chord_progression, prompt_duration, musicgen_model, num_iterations, bpm], outputs=output_audio)
 
-iface.launch()
+iface.launch()

requirements.txt

@@ -1,6 +1,6 @@
-torch==2.1.0
-torchaudio
+#torch==2.1.0
+#torchaudio
 audiocraft
 musiclang_predict
 pyFluidSynth 
-midi2audio 
+midi2audio