Update inference_app.py

inference_app.py

@@ -6,18 +6,22 @@ import gradio as gr
 
 from gradio_molecule3d import Molecule3D
 
+from inference_single import inference
+
 
 
 
 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
+    inference(input_protein_1, input_protein_2)
+
     # return an output pdb file with the protein and two chains A and B.  
     # also return a JSON with any metrics you want to report
     metrics = {"mean_plddt": 80, "binding_affinity": 2}
     end_time = time.time()
     run_time = end_time - start_time
-    return "test_out.pdb",json.dumps(metrics), run_time
+    return "output.pdb",json.dumps(metrics), run_time
 
 with gr.Blocks() as app: