import gradio as gr
import jax

demo = gr.Blocks()
def speech_to_text():
    return "Test"
with demo:
    text = gr.Textbox()
    seqs= ["DSDAJSKDPAKSDPKPEKPKEPKDPAD", "MKTPKPSKPDKAPSKFKRNGKMSODAPS;DPALSPFKOIFMENENOLMAPSDASPFMKOVMFEOM"]
    seqChoice = gr.Radio(seqs)

    b1 = gr.Button("Run ProtGPT2")
    b2 = gr.Button("Predicted structure")

    b1.click(speech_to_text, inputs=text, outputs=seqs)

demo.launch()