Update inference_app.py

inference_app.py

@@ -21,8 +21,8 @@ def set_all_to_zero(input_pdb_file_1, input_pdb_file_2, output_file):
     file.write(output_file)
 
 
-# def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2,  input_protein_2):
-def predict(input_protein_1, input_protein_2):
+def predict(input_seq_1, input_msa_1, input_protein_1, input_seq_2, input_msa_2,  input_protein_2):
+# def predict(input_protein_1, input_protein_2):
     start_time = time.time()
 
     # Do inference here
@@ -65,11 +65,15 @@ with gr.Blocks() as app:
     gr.Examples(
         [
             [
+                "",
+                "",
                 "3v1c_A.pdb",
+                "",
+                "",
                 "3v1c_B.pdb",
             ],
         ],
-        [input_protein_1, input_protein_2],
+        [input_seq_1, input_msa_1, input_protein_1, input_seq_2, input_msa_2,  input_protein_2],
     )
     reps =    [
     {
@@ -105,6 +109,6 @@ with gr.Blocks() as app:
     metrics = gr.JSON(label="Metrics")
     run_time = gr.Textbox(label="Runtime")
 
-    btn.click(predict, inputs=[input_protein_1, input_protein_2], outputs=[out, metrics, 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, metrics, run_time])
 
 app.launch()