Update inference_app.py

inference_app.py

@@ -24,10 +24,11 @@ def set_all_to_zero(input_pdb_file_1, input_pdb_file_2, 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):
     start_time = time.time()
+
     # Do inference here
-    
     # return an output pdb file with the protein and two chains A and B.  
     output_file = "test_out.pdb"
+    print(input_protein_1, input_protein_2)
     set_all_to_zero(input_protein_1, input_protein_2, output_file)
     # also return a JSON with any metrics you want to report
     metrics = {"F_nat": 100}
@@ -45,7 +46,7 @@ with gr.Blocks() as app:
         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 (A3M)")
-            input_protein_1 = gr.File(label="Input Protein 2 monomer (PDB)")
+            input_protein_1 = gr.File(label="Input Protein 1 monomer (PDB)")
         with gr.Column():
             # input_seq_2 = gr.Textbox(lines=3, label="Input Protein 2 sequence (FASTA)")
             # input_msa_2 = gr.File(label="Input MSA Protein 2 (A3M)")
@@ -67,7 +68,6 @@ with gr.Blocks() as app:
             [
                 "3v1c_A.pdb",
                 "3v1c_B.pdb",
-                
             ],
         ],
         [input_protein_1, input_protein_2],