update url

app.py

@@ -14,7 +14,7 @@ def model_load():
     config = OmegaConf.load("config.yaml")
     wrapper = TransformerWrapper(config)
     wrapper = wrapper.load_from_checkpoint(
-        "https://huggingface.co/sweetcocoa/pop2piano/raw/main/model-1999-val_0.67311615.ckpt",
+        "https://huggingface.co/sweetcocoa/pop2piano/resolve/main/model-1999-val_0.67311615.ckpt",
         config=config,
     ).cuda()
     model_id = "dpipqxiy"
@@ -27,6 +27,7 @@ def main():
     wrapper, model_id, config = model_load()
     composers = list(config.composer_to_feature_token.keys())
     dest_dir = "ytsamples"
+    os.makedirs(dest_dir, exist_ok=True)
     composer = st.selectbox(label="Arranger", options=composers)
     file_up = st.file_uploader("Upload an audio", type=["mp3", "wav"])
 
@@ -47,7 +48,6 @@ def main():
                     show_plot=False,
                     save_midi=True,
                     save_mix=True,
-                    vqvae=None,
                 )
 
             with open(midi_path, "rb") as midi_f: