Update utils.py

utils.py

@@ -18,7 +18,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model, tokenizer = get_model_and_tokenizer()
 
 
-def create_seed_string(genre: str = "OTHER") -> str:
+def create_seed_string(genre: str = "OTHER", artist: str = "OTHER") -> str:
     """
     Creates a seed string for generating a new piece.
 
@@ -28,10 +28,14 @@ def create_seed_string(genre: str = "OTHER") -> str:
     Returns:
         str: The seed string.
     """
-    if genre == "RANDOM":
+    if genre == "RANDOM" and artist == "RANDOM":
         seed_string = "PIECE_START"
+    elif genre == "RANDOM" and artist != "RANDOM":
+        seed_string = f"PIECE_START GENRE=RANDOM GENRE={artist} TRACK_START"
+    elif genre != "RANDOM" and artist == "RANDOM":
+        seed_string = f"PIECE_START GENRE={genre} GENRE=RANDOM TRACK_START"
     else:
-        seed_string = f"PIECE_START GENRE={genre} TRACK_START"
+        seed_string = f"PIECE_START GENRE={genre} GENRE={artist} TRACK_START"
     return seed_string
 
 
@@ -235,11 +239,9 @@ def generate_song(
                                                     instruments string, generated song string, and number of tokens string.
     """
     if text_sequence == "":
-        seed = (' ').join((genre, artist))
-        seed_string = create_seed_string(seed)
+        seed_string = create_seed_string(genre, artist)
     else:
         seed_string = text_sequence
-    print(seed_string)
 
     generated_sequence = generate_new_instrument(seed=seed_string, temp=temp)
     audio, midi_file, fig, instruments_str, num_tokens = get_outputs_from_string(