Update app.py

app.py

@@ -261,7 +261,7 @@ def load_input(song_path, arranger_id):
 
 def download_piano(youtube_link):
     yt = pytube.YouTube(youtube_link)
-    download_path = os.path.join(yt_dir, yt.title + ".mp4")
+    download_path = os.path.join(yt_dir, f"{binascii.hexlify(os.urandom(8)).decode()}.mp4")
     yt.streams.filter(only_audio=True).first().download(filename=download_path)
 
     # convert to mp3
@@ -290,11 +290,12 @@ def inference(yt_link, arranger_id):
 
 def post_process(generated):
     print("post processing")
-    midi = tokenizer.decode(generated.argmax(dim=-1).cpu())
+    print(generated.argmax(dim=-1).shape)
+    midi = tokenizer.decode(generated.argmax(dim=-1).unsqueeze(0).cpu())
 
     # random name
     output_midi_path = os.path.join(midi_dir, f"{binascii.hexlify(os.urandom(8)).decode()}.mid")
-    midi.dump_midi(os.path.join(midi_dir, output_midi_path))
+    midi.dump_midi(output_midi_path)
 
     print("exporting")
     return output_midi_path