Spaces:
Running
Running
small fix
Browse files- inference_app.py +4 -3
inference_app.py
CHANGED
|
@@ -7,7 +7,7 @@ from run_on_seq import run_on_sample_seqs
|
|
| 7 |
from env_consts import RUN_CONFIG_PATH, OUTPUT_PROT_PATH, OUTPUT_LIG_PATH, MODEL_NAME_TO_CKPT
|
| 8 |
|
| 9 |
|
| 10 |
-
def predict(input_sequence, input_ligand,
|
| 11 |
print("Strating inference!!!!!!!!!!!!!!!!!", input_sequence, input_ligand, input_protein)
|
| 12 |
start_time = time.time()
|
| 13 |
# Do inference here
|
|
@@ -23,6 +23,7 @@ def predict(input_sequence, input_ligand, input_msa, input_protein, model_variat
|
|
| 23 |
|
| 24 |
|
| 25 |
with gr.Blocks() as app:
|
|
|
|
| 26 |
gr.Markdown("DockFormer")
|
| 27 |
|
| 28 |
model_variation = gr.Dropdown(
|
|
@@ -36,7 +37,7 @@ with gr.Blocks() as app:
|
|
| 36 |
input_sequence = gr.Textbox(lines=3, label="Input Protein sequence (FASTA)")
|
| 37 |
input_ligand = gr.Textbox(lines=3, label="Input ligand SMILES")
|
| 38 |
with gr.Row():
|
| 39 |
-
input_msa = gr.File(label="Input Protein MSA (A3M)")
|
| 40 |
input_protein = gr.File(label="Input protein monomer")
|
| 41 |
|
| 42 |
# define any options here
|
|
@@ -76,7 +77,7 @@ with gr.Blocks() as app:
|
|
| 76 |
metrics = gr.JSON(label="Metrics")
|
| 77 |
run_time = gr.Textbox(label="Runtime")
|
| 78 |
|
| 79 |
-
btn.click(predict, inputs=[input_sequence, input_ligand,
|
| 80 |
outputs=[out, metrics, run_time])
|
| 81 |
|
| 82 |
app.launch()
|
|
|
|
| 7 |
from env_consts import RUN_CONFIG_PATH, OUTPUT_PROT_PATH, OUTPUT_LIG_PATH, MODEL_NAME_TO_CKPT
|
| 8 |
|
| 9 |
|
| 10 |
+
def predict(input_sequence, input_ligand, input_protein, model_variation):
|
| 11 |
print("Strating inference!!!!!!!!!!!!!!!!!", input_sequence, input_ligand, input_protein)
|
| 12 |
start_time = time.time()
|
| 13 |
# Do inference here
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
with gr.Blocks() as app:
|
| 26 |
+
print("Starting app!!!!")
|
| 27 |
gr.Markdown("DockFormer")
|
| 28 |
|
| 29 |
model_variation = gr.Dropdown(
|
|
|
|
| 37 |
input_sequence = gr.Textbox(lines=3, label="Input Protein sequence (FASTA)")
|
| 38 |
input_ligand = gr.Textbox(lines=3, label="Input ligand SMILES")
|
| 39 |
with gr.Row():
|
| 40 |
+
# input_msa = gr.File(label="Input Protein MSA (A3M)")
|
| 41 |
input_protein = gr.File(label="Input protein monomer")
|
| 42 |
|
| 43 |
# define any options here
|
|
|
|
| 77 |
metrics = gr.JSON(label="Metrics")
|
| 78 |
run_time = gr.Textbox(label="Runtime")
|
| 79 |
|
| 80 |
+
btn.click(predict, inputs=[input_sequence, input_ligand, input_protein, model_variation],
|
| 81 |
outputs=[out, metrics, run_time])
|
| 82 |
|
| 83 |
app.launch()
|