Update inference_app.py

inference_app.py

@@ -1,5 +1,6 @@
 
 import time
+import json 
 
 import gradio as gr
 
@@ -12,9 +13,11 @@ def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2,
     start_time = time.time()
     # Do inference here
     # 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", run_time
+    return "test_out.pdb",json.dumps(metrics), run_time
 
 with gr.Blocks() as app:
 
@@ -82,10 +85,12 @@ with gr.Blocks() as app:
       "color": "greenCarbon"
     }      
   ]
+    # outputs 
     
     out = Molecule3D(reps=reps)
+    metrics = gr.JSON(label="Metrics")
     run_time = gr.Textbox(label="Runtime")
 
-    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])
+    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()