Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import allin1
 import zipfile
+import shutil
 from subprocess import Popen
 from concurrent.futures import ThreadPoolExecutor
 
@@ -70,6 +71,10 @@ def compress_files(demucs_output_path, dissector_file, original_audio):
     with ThreadPoolExecutor(max_workers=8) as executor:
         list(executor.map(convert_to_mp3, wav_files, mp3_files))
 
+    try:
+        shutil.rmtree(BASIC_PITCH_DIR)
+    except:  # FileNotFoundError on first run
+        pass
     if not os.path.exists(BASIC_PITCH_DIR):
         os.makedirs(BASIC_PITCH_DIR)
     command = [