Update app.py

app.py

@@ -205,7 +205,9 @@ def generate_music(prompt: str, audio_length: int):
         musicgen_model, musicgen_processor = get_musicgen_model(model_key)
 
         device = "cuda" if torch.cuda.is_available() else "cpu"
-        inputs = musicgen_processor(text=[prompt], padding=True, return_tensors="pt").to(device)
+        # Process the input and move each tensor to the proper device
+        inputs = musicgen_processor(text=[prompt], padding=True, return_tensors="pt")
+        inputs = {k: v.to(device) for k, v in inputs.items()}
 
         with torch.inference_mode():
             outputs = musicgen_model.generate(**inputs, max_new_tokens=audio_length)