updated description and cleanup

playground.py

@@ -3,7 +3,7 @@ from load import LoadModel
 from generate import GenerateMidiText
 from constants import INSTRUMENT_TRANSFER_CLASSES
 from decoder import TextDecoder
-from utils import get_miditok, index_has_substring
+from utils import get_miditok
 from playback import get_music
 from matplotlib import pylab
 import sys
@@ -17,8 +17,6 @@ sys.modules["pylab"] = pylab
 
 model_repo = "JammyMachina/elec-gmusic-familized-model-13-12__17-35-53"
 n_bar_generated = 8
-# model_repo = "JammyMachina/improved_4bars-mdl"
-# n_bar_generated = 4
 
 model, tokenizer = LoadModel(
     model_repo,
@@ -145,8 +143,6 @@ def instrument_col(default_inst, col_id):
         regenerate = gr.State(
             value=False
         )  # initial state should be to generate (not regenerate)
-        # add_bars = gr.Checkbox(value=False, label="Add Bars")
-        # add_bar_count = gr.Dropdown([1, 2, 4, 8], value=1, label="Add Bars")
         gen_btn = gr.Button("Generate")
         inst_audio = gr.Audio(label="TRACK Audio", show_label=True)
         output_txt = gr.Textbox(
@@ -181,7 +177,7 @@ with gr.Blocks() as demo:
         """
         For each **TRACK**, choose your **instrument** along with **creativity** (temperature) and **note density**. 
         Then, hit the **Generate** Button, and after a few seconds a track should have been generated. 
-        Check the **piano roll** and listen to the TRACK! If you don't like the track, hit the generate button to regenerate it. 
+        Check the **piano roll** and listen to the TRACK! If you don't like it, hit the generate button to regenerate it. 
         You can then generate more tracks and listen to the **mixed audio**! \n
         Does it sound nice? Maybe a little robotic and laking some depth... Well, you can download the MIDI file and import it in your favorite DAW to edit the instruments and add some effects!
         """