Update app.py

app.py

@@ -131,25 +131,43 @@ def generation(text):
     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, 
-  inputs=["text"],
-  outputs=['audio'],
-  title=title,
+#title = "Music gen with GPT-2"
+#title2 = "Midi token generation"
+
+#iface = gr.Interface(
+#  fn=process, 
+#  inputs=["text"],
+#  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."
-)
-
-iface2 = 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)
-iface2.launch(debug=True)
+#  article="This demo is inspired in the notebook from https://huggingface.co/TristanBehrens/js-fakes-4bars."
+#)
+
+#iface2 = gr.Interface(
+#    fn=generation,
+#    inputs=["text"],
+#    outputs=["text"],
+#    title=title2,
+#    article="This will generate the encoded text that represents Midi language."
+#)
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("Midi Generation")
+    with gr.Tab("Token generation"):
+        text_input = gr.Textbox()
+        text_output = gr.Textbox()
+        text_button = gr.button("generate tokens")
+    with gr.Tab("Audio generation"):
+        text_input = gr.Textbox()
+        audio_output = gr.Audio()
+        audio_button = gr.button("generate audio")
+    
+    text_button.click(generation, inputs=text_input, outputs=text_output)
+    audio_button.click(process, inputs=text_input, outputs=audio_output)
+
+
+if __name__ == "__main__":
+    demo.launch(
+
+#iface.launch(debug=True)