Update utils.py

utils.py

@@ -213,6 +213,7 @@ def change_tempo(
 
 def generate_song(
     genre: str = "OTHER",
+    artist: str = "KATE_BUSH",
     temp: float = 0.75,
     text_sequence: str = "",
     qpm: int = 120,
@@ -224,6 +225,7 @@ def generate_song(
         model (AutoModelForCausalLM): The pretrained model used for generating the sequences.
         tokenizer (AutoTokenizer): The tokenizer used to encode and decode the sequences.
         genre (str, optional): The genre of the song. Defaults to "OTHER".
+        artist (str, optional): The artist style to inspire the song. Defaults to "KATE_BUSH".
         temp (float, optional): The temperature for the generation, which controls the randomness. Defaults to 0.75.
         text_sequence (str, optional): The initial text sequence for the song. Defaults to "".
         qpm (int, optional): The quarter notes per minute. Defaults to 120.
@@ -233,7 +235,8 @@ def generate_song(
                                                     instruments string, generated song string, and number of tokens string.
     """
     if text_sequence == "":
-        seed_string = create_seed_string(genre)
+        seed = (' ').join((genre, artist))
+        seed_string = create_seed_string(seed)
     else:
         seed_string = text_sequence