simonduerr's picture
Upload folder using huggingface_hub
7860afc verified
raw
history blame
1.21 kB
import gradio as gr
from gradio_cofoldinginput import CofoldingInput
import json
def predict(input):
input = json.dumps(input)
return input
with gr.Blocks() as demo:
jobname = gr.Textbox(label="Job Name")
inp=CofoldingInput(label="Input")
preinput = {"chains": [
{
"class": "DNA",
"sequence": "ATGCGT",
"chain": "A",
"msa": True
},
{
"class": "protein",
"sequence": "AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA",
"chain": "B",
"msa": True
},
{
"class": "ligand",
"name": "ZN",
"smiles": "",
"sdf": "",
"chain": "C"
},
{
"class": "ligand",
"smiles": "CCCCCCCCCCCCCCCCCCCC",
"name": "",
"sdf": "",
"chain": "D"
}
], "covMods":[]
}
# inp2=CofoldingInput(preinput, label="Input prefilled")
btn = gr.Button("Submit")
out = gr.HTML()
gr.Examples([["test",preinput]], inputs=[jobname,inp])
btn.click(predict, inputs=[inp], outputs=[out])
if __name__ == "__main__":
demo.launch()