OleinikovasV commited on
Commit
e256226
·
verified ·
1 Parent(s): 025b93f

Update inference_app.py

Browse files
Files changed (1) hide show
  1. inference_app.py +2 -2
inference_app.py CHANGED
@@ -71,7 +71,7 @@ def predict(input_sequence, input_ligand, input_msa, input_protein):
71
 
72
  mol_coords = mol.GetConformer().GetPositions()
73
  # new_coord = [0, 0, 0]
74
- new_coord = np.mean(mol_coords, axis=0) + [6.05, 6.05, 6.05]
75
  output_file = "test_out.pdb"
76
  set_protein_to_new_coord(input_protein, new_coord, output_file)
77
 
@@ -120,7 +120,7 @@ with gr.Blocks() as app:
120
  reps = [
121
  {
122
  "model": 0,
123
- "style": "stick",
124
  "color": "whiteCarbon",
125
  },
126
  {
 
71
 
72
  mol_coords = mol.GetConformer().GetPositions()
73
  # new_coord = [0, 0, 0]
74
+ new_coord = np.mean(mol_coords, axis=0) + [3.5, 3.5, 3.5]
75
  output_file = "test_out.pdb"
76
  set_protein_to_new_coord(input_protein, new_coord, output_file)
77
 
 
120
  reps = [
121
  {
122
  "model": 0,
123
+ "style": "vdW",
124
  "color": "whiteCarbon",
125
  },
126
  {