Update app.py

app.py

@@ -225,19 +225,19 @@ class Mustango:
 
 
 # Initialize Mustango
-# mustango = Mustango(device="cpu")
-# mustango.vae.to(device_type)
-# mustango.stft.to(device_type)
-# mustango.model.to(device_type)
-# mustango.music_model.beats_model.to(device_type)
-# mustango.music_model.chords_model.to(device_type)
+mustango = Mustango(device="cpu")
+mustango.vae.to(device_type)
+mustango.stft.to(device_type)
+mustango.model.to(device_type)
+mustango.music_model.beats_model.to(device_type)
+mustango.music_model.chords_model.to(device_type)
         
 # if torch.cuda.is_available():
 #     mustango = Mustango(device=device_selection)
 # else:
 #     mustango = Mustango(device="CPU")
 
-mustango = Mustango(device=device_selection)
+# mustango = Mustango(device=device_selection)
 
 mustango.model.device = device_selection