Update inference_app.py

inference_app.py

@@ -8,7 +8,7 @@ from gradio_molecule3d import Molecule3D
 
 
 
-def predict (input_seq_1, input_protein_1, input_seq_2,  input_protein_2):
+def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2,  input_protein_2):
     start_time = time.time()
     # Do inference here
     # return an output pdb file with the protein and two chains A and B.  
@@ -24,9 +24,11 @@ with gr.Blocks() as app:
     with gr.Row():
         with gr.Column():
             input_seq_1 = gr.Textbox(lines=3, label="Input Protein 1 sequence (FASTA)")
+            input_msa_1 = gr.File(label="Input MSA Protein 1 (FASTA)")
             input_protein_1 = gr.File(label="Input Protein 2 monomer (PDB)")
         with gr.Column():
-            input_seq_2 = gr.Textbox(lines=3, label="Input Protein 1 sequence (FASTA)")
+            input_seq_2 = gr.Textbox(lines=3, label="Input Protein 2 sequence (FASTA)")
+            input_msa_2 = gr.File(label="Input MSA Protein 2 (FASTA)")
             input_protein_2 = gr.File(label="Input Protein 2 structure (PDB)")
         
         
@@ -84,6 +86,6 @@ with gr.Blocks() as app:
     out = Molecule3D(reps=reps)
     run_time = gr.Textbox(label="Runtime")
 
-    btn.click(predict, inputs=[input_seq_1, input_protein_1, input_seq_2,  input_protein_2], outputs=[out, run_time])
+    btn.click(predict, inputs=[input_seq_1, input_msa_1, input_protein_1, input_seq_2, input_msa_2,  input_protein_2], outputs=[out, run_time])
 
 app.launch()