bshor commited on
Commit
3f88de2
·
1 Parent(s): d76e1ad
Files changed (1) hide show
  1. 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, input_msa, input_protein, model_variation):
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, input_msa, input_protein, model_variation],
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()