Update inference_app.py

inference_app.py

@@ -54,7 +54,7 @@ def generate_input_conformer(
 
 def set_protein_to_new_coord(input_pdb_file, new_coord, output_file):
     structure = PDBFile.read(input_pdb_file).get_structure()
-    structure.coord = np.array([new_coord] * len(structure.coord))
+    structure.coord = np.ones_like(structure.coord) * np.array(new_coord)
     file = PDBFile()
     file.set_structure(structure)
     file.write(output_file)