Sukanyaaa commited on
Commit
3680967
·
1 Parent(s): 3c186a4

fix inference_app.py

Browse files
Files changed (1) hide show
  1. inference_app.py +2 -2
inference_app.py CHANGED
@@ -391,7 +391,7 @@ def update_pdb_coordinates_from_tensor(input_filename, output_filename, coordina
391
  # atom.set_name(original_name)
392
  # atom.set_leve
393
 
394
- output_filename = "./" + output_filename
395
  # Save the updated structure to a new PDB file
396
  io = PDBIO()
397
  io.set_structure(structure)
@@ -409,7 +409,7 @@ def merge_pdb_files(file1, file2, output_file):
409
  - file2 (str): Path to the second PDB file (e.g., ligand).
410
  - output_file (str): Path to the output file where the merged structure will be saved.
411
  """
412
- output_file = "./" + output_file
413
  with open(output_file, 'w') as outfile:
414
  # Copy the contents of the first file
415
  with open(file1, 'r') as f1:
 
391
  # atom.set_name(original_name)
392
  # atom.set_leve
393
 
394
+ output_filename = "/tmp/" + output_filename
395
  # Save the updated structure to a new PDB file
396
  io = PDBIO()
397
  io.set_structure(structure)
 
409
  - file2 (str): Path to the second PDB file (e.g., ligand).
410
  - output_file (str): Path to the output file where the merged structure will be saved.
411
  """
412
+ output_file = "/tmp/" + output_file
413
  with open(output_file, 'w') as outfile:
414
  # Copy the contents of the first file
415
  with open(file1, 'r') as f1: