fix map loc

app.py

@@ -102,7 +102,7 @@ def generate_music(
         vocab_size=128,
     )
     model = TunesFormer(patch_config, char_config, share_weights=SHARE_WEIGHTS)
-    checkpoint = torch.load(weights, map_location=torch.device("cpu"))
+    checkpoint = torch.load(weights, map_location=DEVICE)
     model.load_state_dict(checkpoint["model"])
     model = model.to(DEVICE)
     model.eval()