Update app.py

app.py

@@ -124,7 +124,14 @@ def process(text):
     return SAMPLE_RATE, int16_data
 
 
+def generation(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])
+    return generated_sequence
+
 title = "Music gen with GPT-2"
+title2 = "Midi token generation"
 
 iface = gr.Interface(
   fn=process, 
@@ -135,4 +142,12 @@ iface = gr.Interface(
   article="This demo is inspired in the notebook from https://huggingface.co/TristanBehrens/js-fakes-4bars."
 )
 
+iface = gr.Interface(
+    fn=generation
+    inputs=["text"],
+    outputs=["text"],
+    title=title2,
+    article="This will generate the encoded text that represents Midi language."
+)
+
 iface.launch(debug=True)