Update inference_app.py
Browse files- inference_app.py +2 -2
inference_app.py
CHANGED
|
@@ -8,11 +8,11 @@ from inference_base import inference
|
|
| 8 |
def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
|
| 9 |
start_time = time.time()
|
| 10 |
# Do inference here
|
| 11 |
-
inference(input_protein_1, input_protein_2)
|
| 12 |
|
| 13 |
# return an output pdb file with the protein and two chains A and B.
|
| 14 |
# also return a JSON with any metrics you want to report
|
| 15 |
-
metrics = {"mean_plddt": 80, "binding_affinity": 2}
|
| 16 |
end_time = time.time()
|
| 17 |
run_time = end_time - start_time
|
| 18 |
return "output.pdb",json.dumps(metrics), run_time
|
|
|
|
| 8 |
def predict (input_seq_1, input_msa_1, input_protein_1, input_seq_2,input_msa_2, input_protein_2):
|
| 9 |
start_time = time.time()
|
| 10 |
# Do inference here
|
| 11 |
+
metrics = inference(input_protein_1, input_protein_2)
|
| 12 |
|
| 13 |
# return an output pdb file with the protein and two chains A and B.
|
| 14 |
# also return a JSON with any metrics you want to report
|
| 15 |
+
# metrics = {"mean_plddt": 80, "binding_affinity": 2}
|
| 16 |
end_time = time.time()
|
| 17 |
run_time = end_time - start_time
|
| 18 |
return "output.pdb",json.dumps(metrics), run_time
|