Update app.py

app.py

@@ -20,8 +20,9 @@ def load_model():
 def generate_music_tensors(descriptions, duration: int, device):
     # Load the model and move it to the specified device
     model = load_model()
-    model = model.to(device)
+    model = model
 
+    
     model.set_generation_params(
         use_sampling=True,
         top_k=250,