OleinikovasV commited on
Commit
e50294a
·
verified ·
1 Parent(s): 7571d02

Update inference_app.py

Browse files
Files changed (1) hide show
  1. inference_app.py +6 -5
inference_app.py CHANGED
@@ -21,7 +21,8 @@ def set_all_to_zero(input_pdb_file_1, input_pdb_file_2, output_file):
21
  file.write(output_file)
22
 
23
 
24
- def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
 
25
  start_time = time.time()
26
  # Do inference here
27
 
@@ -42,12 +43,12 @@ with gr.Blocks() as app:
42
  gr.Markdown("Title, description, and other information about the model")
43
  with gr.Row():
44
  with gr.Column():
45
- input_seq_1 = gr.Textbox(lines=3, label="Input Protein 1 sequence (FASTA)")
46
- input_msa_1 = gr.File(label="Input MSA Protein 1 (A3M)")
47
  input_protein_1 = gr.File(label="Input Protein 2 monomer (PDB)")
48
  with gr.Column():
49
- input_seq_2 = gr.Textbox(lines=3, label="Input Protein 2 sequence (FASTA)")
50
- input_msa_2 = gr.File(label="Input MSA Protein 2 (A3M)")
51
  input_protein_2 = gr.File(label="Input Protein 2 structure (PDB)")
52
 
53
 
 
21
  file.write(output_file)
22
 
23
 
24
+ # def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
25
+ def predict(input_protein_1, input_protein_2):
26
  start_time = time.time()
27
  # Do inference here
28
 
 
43
  gr.Markdown("Title, description, and other information about the model")
44
  with gr.Row():
45
  with gr.Column():
46
+ # input_seq_1 = gr.Textbox(lines=3, label="Input Protein 1 sequence (FASTA)")
47
+ # input_msa_1 = gr.File(label="Input MSA Protein 1 (A3M)")
48
  input_protein_1 = gr.File(label="Input Protein 2 monomer (PDB)")
49
  with gr.Column():
50
+ # input_seq_2 = gr.Textbox(lines=3, label="Input Protein 2 sequence (FASTA)")
51
+ # input_msa_2 = gr.File(label="Input MSA Protein 2 (A3M)")
52
  input_protein_2 = gr.File(label="Input Protein 2 structure (PDB)")
53
 
54