Update app.py

app.py

@@ -17,11 +17,9 @@ def load_model():
     return model
 
 def generate_music_tensors(descriptions, duration: int):
-    model = load_model()
-
-    # Manually set the device to CPU
+    # Create a new instance of the model with the desired device
     device = torch.device('cpu')
-    model = model.to(device)
+    model = MusicGen.get_pretrained('facebook/musicgen-small').to(device)
 
     model.set_generation_params(
         use_sampling=True,