wangjin2000 commited on
Commit
19d3dce
·
verified ·
1 Parent(s): 6ead239

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -346,8 +346,9 @@ def predict_peptide_from_file(base_model_path, finetuned_model_path, file_obj, m
346
 
347
  #get metrics for ground truth peptide
348
  ppl = compute_pseudo_perplexity(loaded_model, tokenizer, protein_seq, peptide_seq)
349
- if plddt_iptm_yes=="yes":
350
- plddt, iptm = compute_plddt_iptm(protein_seq, peptide_seq) #too time-consuming
 
351
  else:
352
  plddt, iptm = [0, 0]
353
 
 
346
 
347
  #get metrics for ground truth peptide
348
  ppl = compute_pseudo_perplexity(loaded_model, tokenizer, protein_seq, peptide_seq)
349
+ if plddt_iptm_yes=="yes":
350
+ plddt, iptm = get_plddt(protein_seq, generated_binder)
351
+ #plddt, iptm = compute_plddt_iptm(protein_seq, peptide_seq) #too time-consuming
352
  else:
353
  plddt, iptm = [0, 0]
354