Update app.py

app.py

@@ -109,20 +109,21 @@ def empty_note_sequence(qpm=120.0, total_time=0.0):
     return note_sequence
 
 def process(text):
-  input_ids = tokenizer.encode(text, return_tensors="pt")
-  generated_ids = model.generate(input_ids, max_length=500)
-  generated_sequence = tokenizer.decode(generated_ids[0])
+    input_ids = tokenizer.encode(text, return_tensors="pt")
+    generated_ids = model.generate(input_ids, max_length=500)
+    generated_sequence = tokenizer.decode(generated_ids[0])
+
+    # Convert text of notes to audio
+    note_sequence = token_sequence_to_note_sequence(generated_sequence)
+    synth = note_seq.midi_synth.synthesize
+    array_of_floats = synth(note_sequence, sample_rate=SAMPLE_RATE)
+    note_plot = note_seq.plot_sequence(note_sequence, False)
+    array_of_floats /=1.414
+    array_of_floats *= 32767
+    int16_data = array_of_floats.astype(np.int16)
+    return SAMPLE_RATE, int16_data
+
 
-  # Convert text of notes to audio
-  note_sequence = token_sequence_to_note_sequence(generated_sequence)
-  synth = note_seq.midi_synth.synthesize
-  array_of_floats = synth(note_sequence, sample_rate=SAMPLE_RATE)
-  note_plot = note_seq.plot_sequence(note_sequence, False)
-  array_of_floats /=1.414
-  array_of_floats *= 32767
-  int16_data = array_of_floats.astype(np.int16)
-  return SAMPLE_RATE, int16_data
-  
 title = "Music generation with GPT-2"
 
 iface = gr.Interface(
@@ -131,7 +132,7 @@ iface = gr.Interface(
   outputs=['audio'],
   title=title,
   #examples=[["PIECE_START"], ["PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST=48 BAR_START NOTE_ON=61"]],
-  article="This demo is inspired in the notebook from https://huggingface.co/TristanBehrens/js-fakes-4bars"
+  article="This demo is inspired in the notebook from https://huggingface.co/TristanBehrens/js-fakes-4bars. Example: ["PIECE_START"], ["PIECE_START STYLE=JSFAKES GENRE=JSFAKES TRACK_START INST=48 BAR_START NOTE_ON=61"]"
 )
 
 iface.launch(debug=True)