simonduerr commited on
Commit
e4eb08d
1 Parent(s): 92c1221

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -288,7 +288,7 @@ def setup_proteinmpnn(model_name="vanilla—v_48_020", backbone_noise=0.00):
288
  hidden_dim=hidden_dim,
289
  num_encoder_layers=num_layers,
290
  num_decoder_layers=num_layers,
291
- augment_eps=backbone_noise,
292
  k_neighbors=checkpoint["num_edges"],
293
  )
294
  model.to(device)
@@ -434,7 +434,7 @@ def update(
434
  return "Error processing PDB"
435
 
436
  model, device = setup_proteinmpnn(
437
- model_name=model_name, backbone_noise=backbone_noise
438
  )
439
 
440
  if designed_chain == "":
 
288
  hidden_dim=hidden_dim,
289
  num_encoder_layers=num_layers,
290
  num_decoder_layers=num_layers,
291
+ augment_eps=float(backbone_noise),
292
  k_neighbors=checkpoint["num_edges"],
293
  )
294
  model.to(device)
 
434
  return "Error processing PDB"
435
 
436
  model, device = setup_proteinmpnn(
437
+ model_name=model_name, backbone_noise=float(backbone_noise)
438
  )
439
 
440
  if designed_chain == "":