Add better description

main.py

@@ -17,19 +17,20 @@ DESCRIPTION = """
 # 🎵 Multitrack Midi Generator 🎶
 This interactive application uses an AI model to generate music sequences based on a chosen genre and various user inputs.
 
-Features:
-🎼 Select the genre for the music.
-🌡️ Use the "Temperature" slider to adjust the randomness of the music generated (higher values will produce more random outputs).
-⏱️ Adjust the "Tempo" slider to change the speed of the music.
-🎹 Use the buttons to generate a new song from scratch, continue generation with the current settings, remove the last added instrument, regenerate the last added instrument with a new one, or change the tempo of the current song.
-Outputs:
+## Features:
+- 🎼 Select the genre for the music.
+- 🌡️ Use the "Temperature" slider to adjust the randomness of the music generated (higher values will produce more random outputs).
+- ⏱️ Adjust the "Tempo" slider to change the speed of the music.
+- 🎹 Use the buttons to generate a new song from scratch, continue generation with the current settings, remove the last added instrument, regenerate the last added instrument with a new one, or change the tempo of the current song.
+
+## Outputs:
 The app outputs the following:
 
-🎧 The audio of the generated song.
-📁 A MIDI file of the song.
-📊 A plot of the song's sequence.
-🎸 A list of the generated instruments.
-📝 The text sequence of the song.
+- 🎧 The audio of the generated song.
+- 📁 A MIDI file of the song.
+- 📊 A plot of the song's sequence.
+- 🎸 A list of the generated instruments.
+- 📝 The text sequence of the song.
 Enjoy creating your own AI-generated music! 🎵
 """
 

utils.py

@@ -51,15 +51,13 @@ def get_instruments(text_sequence: str) -> List[str]:
 
 
 def generate_new_instrument(
-    seed: str, tokenizer: AutoTokenizer, model: AutoModelForCausalLM, temp: float = 0.75
+    seed: str, temp: float = 0.75
 ) -> str:
     """
     Generates a new instrument sequence from a given seed and temperature.
 
     Args:
         seed (str): The seed string for the generation.
-        tokenizer (PreTrainedTokenizer): The tokenizer used to encode and decode the sequences.
-        model (PreTrainedModel): The pretrained model used for generating the sequences.
         temp (float, optional): The temperature for the generation, which controls the randomness. Defaults to 0.75.
 
     Returns:
@@ -209,8 +207,6 @@ def change_tempo(
 
 
 def generate_song(
-    model: AutoModelForCausalLM = model,
-    tokenizer: AutoTokenizer = tokenizer,
     genre: str = "OTHER",
     temp: float = 0.75,
     text_sequence: str = "",
@@ -237,7 +233,7 @@ def generate_song(
         seed_string = text_sequence
 
     generated_sequence = generate_new_instrument(
-        seed=seed_string, tokenizer=tokenizer, model=model, temp=temp
+        seed=seed_string, temp=temp
     )
     audio, midi_file, fig, instruments_str, num_tokens = get_outputs_from_string(
         generated_sequence, qpm