Update app.py

app.py

@@ -176,7 +176,7 @@ class Mustango:
             main_config["scheduler_name"],
             unet_model_config_path=f"{path}/configs/music_diffusion_model_config.json",
         ).to(device)
-        self.model.device = device
+        # self.model.device = device
 
         vae_weights = torch.load(
             f"{path}/vae/pytorch_model_vae.bin", map_location=device
@@ -226,11 +226,11 @@ class Mustango:
 
 # Initialize Mustango
 mustango = Mustango(device="cpu")
-mustango.vae.to(device_selection)
-mustango.stft.to(device_selection)
-mustango.model.to(device_selection)
-mustango.music_model.beats_model.to(device_selection)
-mustango.music_model.chords_model.to(device_selection)
+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()
 # else: