Update app.py

app.py

@@ -31,7 +31,7 @@ description = """
 ## ℹ️ How to use this demo?
 1. Enter the control codes to set the musical form of the generated music. For details, please refer to the below "Control Codes" section. (optional)
 2. Enter the prompt of the generated music. You can set the control codes to set the musical form, ABC header (i.e., note length, tempo, meter, and key) and the motif of the melody. (optional)
-2. You can set the parameters (i.e., number of tunes, maximum patch, top-p, top-k, temperature, random seed and show control code) for the generation. (optional)
+2. You can set the parameters (i.e., number of tunes, maximum patch, top-p, top-k, temperature and random seed) for the generation. (optional)
 3. Click "Submit" and wait for the result.
 4. The generated ABC notation can be played or edited using [EasyABC]([ABC Sheet Music Editor - EasyABC](https://easyabc.sourceforge.net/), you can also use this [Online ABC Player](https://abc.rectanglered.com/) to render the tune.
 
@@ -48,13 +48,14 @@ Inspired by [CTRL](https://huggingface.co/ctrl), we incorporate control codes in
   ```
   where $|c|$ and $|p|$ are the string lengths of the two sections. It is discretized into 11 levels, ranging from no match at all to an exact match (e.g., `E:0` for no similarity, and `E:10` for an exact match).
 
-## ❕Notice
-- If the prefix is given, you must also provide the control codes.
-- The order of the ABC header cannot be changed (i.e., note length, tempo, meter, and key), and also do not support other headers (e.g., composer, title, etc.) as they are not used in the model. For details, please refer to the official [ABC notation](https://abcnotation.com/wiki/abc:standard:v2.1#description_of_information_fields).
-- The demo is based on GPT2-small and trained from scratch on the [Massive ABC Notation Dataset](https://huggingface.co/datasets/sander-wood/massive_abcnotation_dataset).
-- The demo is still in the early stage, and the generated music is not perfect. If you have any suggestions, please feel free to contact me via [email](mailto:shangda@mail.ccom.edu.cn).
+## ❕Caution
+ABC notation is a specialized notation of representing sheet music, and it follows a specific standard format. When interacting with TunesFormer, all trained ABC notation adheres to these standard formats.
 
+If you are unfamiliar with the details of ABC notation, we strongly recommend against manually entering ABC notation. Otherwise, the model may not recognize and generate the music correctly. Inputting incorrect formats might lead to unpredictable outputs or other issues.
 
+A general recommendation is to adjust the desired musical structure and form through control codes and ABC header, rather than directly editing the ABC notation itself.
+
+Please make sure to operate according to the provided formats and guidelines to fully leverage the capabilities of TunesFormer and achieve a satisfying music generation experience.
 """
 
 class Patchilizer:
@@ -305,7 +306,14 @@ class TunesFormer(PreTrainedModel):
         
         return generated_patch, n_seed
 
-def generate_abc(prompt, num_tunes, max_patch, top_p, top_k, temperature, seed, show_control_code):
+def generate_abc(prompt,
+                 num_tunes,
+                 max_patch,
+                 top_p,
+                 top_k,
+                 temperature,
+                 seed,
+                 show_control_code=True):
 
     if torch.cuda.is_available():    
         device = torch.device("cuda")
@@ -419,7 +427,7 @@ M:3/4
 K:D
  de |"D" """
 
-input_prompt = gr.inputs.Textbox(lines=5, label="Prefix", default=default_prompt)
+input_prompt = gr.inputs.Textbox(lines=5, label="Prompt", default=default_prompt)
 input_num_tunes = gr.inputs.Slider(minimum=1, maximum=10, step=1, default=1, label="Number of Tunes")
 input_max_patch = gr.inputs.Slider(minimum=10, maximum=128, step=1, default=128, label="Max Patch")
 input_top_p = gr.inputs.Slider(minimum=0.0, maximum=1.0, step=0.05, default=0.8, label="Top P")