Update inference_app.py

inference_app.py

@@ -71,7 +71,7 @@ def predict(input_sequence, input_ligand, input_msa, input_protein):
 
     mol_coords = mol.GetConformer().GetPositions()    
     # new_coord = [0, 0, 0]
-    new_coord = np.mean(mol_coords, axis=1)
+    new_coord = np.mean(mol_coords, axis=0) + [6.05, 6.05, 6.05]
     output_file = "test_out.pdb"
     set_protein_to_new_coord(input_protein, new_coord, output_file)