Change the theme

app.py

@@ -44,6 +44,7 @@ def prepare_output_file(tokenizer_output):
     wav_output = midi_output.replace(".mid", ".wav")
     midi2audio.FluidSynth().midi_to_audio(midi_output, wav_output)
 
+    from IPython.display import Audio
     return wav_output, wav_output, midi_output
     
 def inference(file_uploaded, composer):
@@ -58,8 +59,9 @@ def inference(file_uploaded, composer):
     return prepare_output_file(tokenizer_output)    
 
 
-block = gr.Blocks()
-
+# Thanks a lot to "https://huggingface.co/Taithrah" for this theme.
+# taken from https://huggingface.co/spaces/NoCrypt/miku
+block = gr.Blocks(theme="Taithrah/Minimal")
 
 with block:
     gr.HTML(
@@ -133,4 +135,5 @@ with block:
         )
 
 block.launch(debug=False)
-shutil.rmtree("./midi_wav_outputs")
+shutil.rmtree("./midi_wav_outputs")
+shutil.rmtree("./yt_dir")