simonduerr commited on
Commit
7b9daf5
1 Parent(s): fde9ec3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -276,7 +276,7 @@ def setup_proteinmpnn(model_name="vanilla—v_48_020", backbone_noise=0.00):
276
  if model_folder_path[-1] != "/":
277
  model_folder_path = model_folder_path + "/"
278
  checkpoint_path = model_folder_path + f"{model_name}.pt"
279
-
280
  checkpoint = torch.load(checkpoint_path, map_location=device)
281
 
282
  noise_level_print = checkpoint["noise_level"]
 
276
  if model_folder_path[-1] != "/":
277
  model_folder_path = model_folder_path + "/"
278
  checkpoint_path = model_folder_path + f"{model_name}.pt"
279
+ print("using ProteinMPNN weights from: ", checkpoint_path)
280
  checkpoint = torch.load(checkpoint_path, map_location=device)
281
 
282
  noise_level_print = checkpoint["noise_level"]