import gradio as gr
from gradio_molecule3d import Molecule3D

# prediction routine
def predict(x):
    return "Model prediction"

reps = [{"model": 0,"style": "cartoon","color": "whiteCarbon"}]

with gr.Blocks() as demo:
    inp = Molecule3D(label="Molecule3D", reps=reps)
    btn = gr.Button("Run")
    out = gr.HTML("")
    btn.click(fn=predict, inputs=[inp], outputs=[out])

demo.launch(ssr_mode=False)